diff --git a/code/submit_train.sh b/code/submit_train.sh
index 533a7f90dad39ad82e8aecd9beb6e46a473a47af..a3ffc45b1706c3d958d24449f50ac6fc83b4cab0 100644
--- a/code/submit_train.sh
+++ b/code/submit_train.sh
@@ -11,6 +11,12 @@
 #SBATCH --output=./slurm_files/slurm-%x-%j.out     # where to write output, %x give job name, %j names job id
 #SBATCH --error=./slurm_files/slurm-%x-%j.err      # where to write slurm error
 
+# load new software stack
+# see https://docs.hpc.gwdg.de/software/software_stacks/index.html
+export PREFERRED_SOFTWARE_STACK=nhr-lmod
+source /sw/etc/profile/profile.sh
+
+
 module load miniconda3
 module load cuda
 source activate dl-gpu # Or whatever you called your environment.