2 gpu scaling

mwalmsley · Nov 7, 2023 · 45d836c · 45d836c
1 parent da24384
commit 45d836c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/only_for_me/narval/train.sh b/only_for_me/narval/train.sh
@@ -1,11 +1,11 @@
 #!/bin/bash
 #SBATCH --time=2:50:0  
 #SBATCH --nodes=1
-#SBATCH --ntasks=4
-#SBATCH --ntasks-per-node=4
+#SBATCH --ntasks=2
+#SBATCH --ntasks-per-node=2
 #SBATCH --cpus-per-task=10
 #SBATCH --mem-per-cpu 4G
-#SBATCH --gres=gpu:v100:4
+#SBATCH --gres=gpu:v100:2
 
 nvidia-smi