Update submit_train.sh to load new software stack

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.