From 7de7e94782f57adc95d1582a280f771cf1c707f5 Mon Sep 17 00:00:00 2001
From: Alexandre Strube <a.strube@fz-juelich.de>
Date: Mon, 29 May 2023 22:20:48 +0200
Subject: [PATCH] call accelerate

---
 src/distrib.slurm | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/distrib.slurm b/src/distrib.slurm
index f203a46..1a86c33 100644
--- a/src/distrib.slurm
+++ b/src/distrib.slurm
@@ -6,7 +6,7 @@
 #SBATCH --cpus-per-task=4
 #SBATCH --output=output.%j
 #SBATCH --error=err.%j
-#SBATCH --time=00:10:00
+#SBATCH --time=00:20:00
 #SBATCH --partition=develbooster
 #SBATCH --gres=gpu:4                                                                                                                                                                                                                           
 
@@ -17,4 +17,4 @@ cd $HOME/2023-may-intro-to-supercompting-jsc/src
 source sc_venv_template/activate.sh
 
 # Run the demo
-time srun python distrib.py
+time srun accelerate launch distrib.py
-- 
GitLab