diff --git a/scripts/benchmark_multi_node.sbatch b/scripts/benchmark_multi_node.sbatch index 75510cf22f9a6fcf1175e82fc1340e2034055ee8..f3dc4becdb90d88ad1f764b53882e4eaf9e1af67 100644 --- a/scripts/benchmark_multi_node.sbatch +++ b/scripts/benchmark_multi_node.sbatch @@ -2,6 +2,8 @@ sbatch <<EOT +#!/bin/bash + #SBATCH -A nsc #SBATCH --nodes=${2} #SBATCH --gres=gpu:8 diff --git a/scripts/benchmark_single_node.sbatch b/scripts/benchmark_single_node.sbatch index 84b5f89ff49c532420eb7dab556a6dc887056744..56be4ecceff4fd97df3eb46318b00af82fad9049 100644 --- a/scripts/benchmark_single_node.sbatch +++ b/scripts/benchmark_single_node.sbatch @@ -2,6 +2,8 @@ sbatch <<EOT +#!/bin/bash + #SBATCH -A nsc #SBATCH --nodes=1 #SBATCH --gres=gpu:${3}