From 664cfb20231dcc22dbfa47b28610f7560fe75b07 Mon Sep 17 00:00:00 2001
From: Cadene <re.cadene@gmail.com>
Date: Thu, 22 Feb 2024 13:04:32 +0000
Subject: [PATCH] Add sbatch.sh

---
 sbatch.sh | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)
 create mode 100644 sbatch.sh

diff --git a/sbatch.sh b/sbatch.sh
new file mode 100644
index 000000000..78cb46062
--- /dev/null
+++ b/sbatch.sh
@@ -0,0 +1,22 @@
+#!/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 --gres=gpu:1         # number of GPUs reserved per node (here 8, or all the GPUs)
+#SBATCH --cpus-per-task=8    # number of cores per task (8x8 = 64 cores, or all the cores)
+#SBATCH --time=02:00:00
+#SBATCH --output=/home/rcadene/slurm/%j.out
+#SBATCH --error=/home/rcadene/slurm/%j.err
+#SBATCH --qos=low
+#SBATCH --mail-user=re.cadene@gmail.com
+#SBATCH --mail-type=ALL
+
+CMD=$@
+echo "command: $CMD"
+
+apptainer exec --nv \
+~/apptainer/nvidia_cuda:12.2.2-devel-ubuntu22.04.sif $SHELL
+
+source ~/.bashrc
+conda activate fowm
+ 
+srun $CMD