From ed0b890e75efcc5251b9be345abeaa07ab509212 Mon Sep 17 00:00:00 2001
From: Jirka Borovec <Borda@users.noreply.github.com>
Date: Thu, 12 Mar 2020 17:36:46 +0100
Subject: [PATCH] update info for GPU (#1021)

---
 .run_local_tests.sh | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/.run_local_tests.sh b/.run_local_tests.sh
index ce7fce58ce..ce2a920819 100644
--- 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*