update info for GPU (#1021)

2020-03-12 17:36:46 +01:00 · 2020-03-12 17:36:46 +01:00 · ed0b890e75
parent b4d4e489bf
commit ed0b890e75
1 changed files with 4 additions and 0 deletions
--- a/.run_local_tests.sh
+++ b/.run_local_tests.sh
@ -1,5 +1,9 @@
 #!/usr/bin/env bash

+# install APEX, see https://github.com/NVIDIA/apex#linux
+# to imitate SLURM set only single node
+export SLURM_LOCALID=0
+
 # use this to run tests
 rm -rf _ckpt_*
 rm -rf ./tests/save_dir*