Merge pull request #37 from argonne-lcf/alcf-haritha-patch-1

Update qsub_polaris.sc
argonne-lcf · Jun 14, 2024 · eac09ed · eac09ed
2 parents c08fbef + b15bc0b
commit eac09ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/learningFrameworks/distributedDeepLearning/DDP/qsub_polaris.sc b/learningFrameworks/distributedDeepLearning/DDP/qsub_polaris.sc
@@ -10,7 +10,7 @@
 
 cd $PBS_O_WORKDIR
 
-module load conda/2023-10-04; conda activate
+module use /soft/modulefiles; module load conda; conda activate
 aprun -n 1 -N 1 --cc depth -d 64 python pytorch_cnn_ddp.py >& pytorch_cnn_ddp.py.1.out
 aprun -n 2 -N 2 --cc depth -d 32 python pytorch_cnn_ddp.py >& pytorch_cnn_ddp.py.2.out
 aprun -n 4 -N 4 --cc depth -d 16 python pytorch_cnn_ddp.py >& pytorch_cnn_ddp.py.4.out