From 833528ef5abde93a507c724b772c924c25397ac3 Mon Sep 17 00:00:00 2001 From: Malachi Date: Tue, 21 Dec 2021 07:38:23 -0700 Subject: [PATCH] Fixes for perlmutter script (#392) --- scripts/nrsqsub_perlmutter | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/scripts/nrsqsub_perlmutter b/scripts/nrsqsub_perlmutter index 9277f2c34..16ccc4159 100755 --- a/scripts/nrsqsub_perlmutter +++ b/scripts/nrsqsub_perlmutter @@ -86,14 +86,15 @@ echo "#SBATCH --gpu-bind=map_gpu:0,1,2,3" >> $SFILE echo "#SBATCH --gpus-per-task=1" >> $SFILE echo "module load PrgEnv-gnu" >>$SFILE +echo "module load cudatoolkit" >>$SFILE echo "module load cpe-cuda" >>$SFILE -echo "module load cuda" >>$SFILE -echo "module load cmake/3.20" >>$SFILE +echo "module load cmake" >>$SFILE echo "module unload cray-libsci" >>$SFILE echo "module list" >>$SFILE echo "nvidia-smi" >>$SFILE echo "export SLURM_CPU_BIND=\"cores\"" >> $SFILE +echo "export CRAY_ACCEL_TARGET=nvidia80" >>$SFILE echo "export MPICH_GPU_SUPPORT_ENABLED=1" >>$SFILE echo "ulimit -s unlimited " >>$SFILE