Fix gpu gres for slurm

2024-02-21 17:03:47 +01:00 · 2024-02-21 17:03:47 +01:00 · f4747bfc9d
commit f4747bfc9d
parent daf22ecdb0
1 changed files with 1 additions and 1 deletions
--- a/experiments/reduce_bench_hemera_gpu.sh
+++ b/experiments/reduce_bench_hemera_gpu.sh
@ -5,7 +5,7 @@
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=32
-#SBATCH --gpus-per-task=1
+#SBATCH --gres=gpu:1
 #SBATCH --mem=24GB
 #SBATCH --output=simulation-%A-%a.out
 #SBATCH --error=simulation-%A-%a.err