run.slurm

#!/bin/bash
#SBATCH -J test
#SBATCH -p wzhdexclu03
#SBATCH -N 4
#SBATCH -n 32
##SBATCH --ntasks-per-node=4
##SBATCH --cpus-per-task=8
#SBATCH --gres=dcu:4
source ~/miniconda3/etc/profile.d/conda.sh
#conda activate torch1.10-dtk22.10-py38

conda activate base
module switch compiler/dtk/22.10
python -u  driver.py #需要运行的程序