single_one_driver.slurm

#!/bin/bash
#SBATCH -J test
#SBATCH -p wzhdexclu03
#SBATCH -N 1
##SBATCH -n 32
#SBATCH --ntasks-per-node=1
#SBATCH --cpus-per-task=32
#SBATCH --gres=dcu:1
#SBATCH -J single_node_1_dcu_train
#SBATCH -o logs/pt-%j.out
#SBATCH -e logs/pt-%j.err
source ~/miniconda3/etc/profile.d/conda.sh
conda activate torch1.10-dtk22.10-py38

#conda activate base
module purge 
module load compiler/devtoolset/7.3.1  mpi/hpcx/gcc-7.3.1  compiler/dtk/23.04
module list


export HIP_VISIBLE_DEVICES=0
python -u  driver.py #需要运行的程序