From 7938adcdfc294c215c8afd0407e1ad594609a11d Mon Sep 17 00:00:00 2001
From: Remi Cadene <re.cadene@gmail.com>
Date: Mon, 4 Mar 2024 22:41:31 +0000
Subject: [PATCH] Add sbatch_hopper.sh

---
 sbatch_hopper.sh | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)
 create mode 100644 sbatch_hopper.sh

diff --git a/sbatch_hopper.sh b/sbatch_hopper.sh
new file mode 100644
index 00000000..cc410048
--- /dev/null
+++ b/sbatch_hopper.sh
@@ -0,0 +1,17 @@
+#!/bin/bash
+#SBATCH --nodes=1            # total number of nodes (N to be defined)
+#SBATCH --ntasks-per-node=1  # number of tasks per node (here 8 tasks, or 1 task per GPU)
+#SBATCH --qos=normal         # number of GPUs reserved per node (here 8, or all the GPUs)
+#SBATCH --partition=hopper-prod
+#SBATCH --gres=gpu:1         # number of GPUs reserved per node (here 8, or all the GPUs)
+#SBATCH --cpus-per-task=12    # number of cores per task
+#SBATCH --mem-per-cpu=11G
+#SBATCH --time=12:00:00
+#SBATCH --output=/admin/home/remi_cadene/slurm/%j.out
+#SBATCH --error=/admin/home/remi_cadene/slurm/%j.err
+#SBATCH --mail-user=remi_cadene@huggingface.co
+#SBATCH --mail-type=ALL
+
+CMD=$@
+echo "command: $CMD"
+srun $CMD