From 4be71a32d1fb8b17947111cc7e8f2bb2ff3b5919 Mon Sep 17 00:00:00 2001 From: Yin Li Date: Thu, 18 Mar 2021 15:17:30 -0400 Subject: [PATCH] Add module and conda env recording --- scripts/example-test.slurm | 4 ++++ scripts/example-train.slurm | 5 +++++ 2 files changed, 9 insertions(+) diff --git a/scripts/example-test.slurm b/scripts/example-test.slurm index 1036450..4e7398b 100644 --- a/scripts/example-test.slurm +++ b/scripts/example-test.slurm @@ -14,8 +14,12 @@ hostname; pwd; date # set computing environment, e.g. with module or anaconda + #module load python +#module list + #source $HOME/anaconda3/bin/activate pytorch_env +#conda info export OMP_NUM_THREADS=$SLURM_CPUS_ON_NODE # use MKL-DNN diff --git a/scripts/example-train.slurm b/scripts/example-train.slurm index 3cd4049..d6cc894 100644 --- a/scripts/example-train.slurm +++ b/scripts/example-train.slurm @@ -26,8 +26,13 @@ hostname; pwd; date # set computing environment, e.g. with module or anaconda + #module load python +#module list + #source $HOME/anaconda3/bin/activate pytorch_env +#conda info + srun m2m.py train \ --train-in-patterns "train/R0-*.npy,train/R1-*.npy" \