hpc

hpc.sh

@@ -0,0 +1,12 @@
+#!/bin/bash
+#SBATCH --job-name=my_gpu_job         # 任务名称，可以根据需要修改
+#SBATCH --partition=gpu_v100s         # 分区名称，与 salloc 命令中一致
+#SBATCH --nodes=1                     # 申请 1 个节点
+#SBATCH --ntasks-per-node=8           # 每个节点运行 8 个任务
+#SBATCH --mem=81920                   # 分配内存（单位为 MB）
+#SBATCH --gres=gpu:1                  # 请求 1 个 GPU 资源
+#SBATCH --time=2:00:00                # 设置最大运行时间为 2 小时
+
+module load cuda/12.1.0
+source .venv/bin/activate
+srun python3 run_all.py