Added CUDA_VISIBLE_DEVICES variable to job scripts

atlas_server/src/static/training/predict.job.sh

@@ -17,6 +17,8 @@ module use ${SOFTWARE_STAGE_MODULES} &> /dev/null
 module load ${ATLAS_MODULE}
 echo Modules loaded
 
+export CUDA_VISIBLE_DEVICES=0,1,2,3
+
 # Environment variables
 export SCRATCH_USER=/p/scratch/cjinm16/schiffer1/
 export OUTPUT_SPACING=16

atlas_server/src/static/training/train.job.sh

@@ -17,6 +17,8 @@ module use ${SOFTWARE_STAGE_MODULES} &> /dev/null
 module load ${ATLAS_MODULE}
 echo Modules loaded
 
+export CUDA_VISIBLE_DEVICES=0,1,2,3
+
 # Environment variables
 export BRAIN=__BRAIN__
 export LABELS=__LABELS__