| #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 |