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