update info for GPU (Lightning-AI#1021)

lezwon · Mar 12, 2020 · ed0b890 · ed0b890
1 parent b4d4e48
commit ed0b890
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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*