From a76d7defc28969c0a38aa3d2d6a9245fd86b9bad Mon Sep 17 00:00:00 2001
From: Christian Schiffer <c.schiffer@fz-juelich.de>
Date: Fri, 26 Mar 2021 07:20:04 +0100
Subject: [PATCH] Added CUDA_VISIBLE_DEVICES variable to job scripts
---
atlas_server/src/static/training/predict.job.sh | 2 ++
atlas_server/src/static/training/train.job.sh | 2 ++
2 files changed, 4 insertions(+)
diff --git a/atlas_server/src/static/training/predict.job.sh b/atlas_server/src/static/training/predict.job.sh
index 1ec14eb..c389015 100644
--- a/atlas_server/src/static/training/predict.job.sh
+++ b/atlas_server/src/static/training/predict.job.sh
@@ -17,6 +17,8 @@ module use ${SOFTWARE_STAGE_MODULES} &> /dev/null
module load ${ATLAS_MODULE}
echo Modules loaded
+export CUDA_VISIBLE_DEVICES=0,1,2,3
+
# Environment variables
export SCRATCH_USER=/p/scratch/cjinm16/schiffer1/
export OUTPUT_SPACING=16
diff --git a/atlas_server/src/static/training/train.job.sh b/atlas_server/src/static/training/train.job.sh
index b5651e9..d0249f9 100644
--- a/atlas_server/src/static/training/train.job.sh
+++ b/atlas_server/src/static/training/train.job.sh
@@ -17,6 +17,8 @@ module use ${SOFTWARE_STAGE_MODULES} &> /dev/null
module load ${ATLAS_MODULE}
echo Modules loaded
+export CUDA_VISIBLE_DEVICES=0,1,2,3
+
# Environment variables
export BRAIN=__BRAIN__
export LABELS=__LABELS__
--
GitLab