File size: 590 Bytes
e2fb081 |
1 2 3 4 5 6 7 8 9 10 11 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 |