diff --git a/scripts/run_benchmark_single_node.sh b/scripts/run_benchmark_single_node.sh
index 1b52ca43c1ba432b2819dfef6574b2bc7b03a475..199430e9523f838a22e6df4cf8f6c87d359df649 100644
--- a/scripts/run_benchmark_single_node.sh
+++ b/scripts/run_benchmark_single_node.sh
@@ -5,7 +5,7 @@ cat <<EOT > scripts/benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_iteration
 
 #SBATCH -A nsc
 #SBATCH --nodes=1
-#SBATCH --gres=gpu:8
+#SBATCH --gres=gpu:${3}
 #SBATCH --time=0-0:10:00
 #SBATCH --output=sbatch_out/benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_iteration${5}.out
 #SBATCH --reservation=devel