Update Distributed training.md

2019-09-26 15:27:34 -04:00 · 2019-09-26 15:27:34 -04:00 · cefcf4cd12
parent e713e2e1e0
commit cefcf4cd12
1 changed files with 3 additions and 0 deletions
--- a/docs/Trainer/Distributed
+++ b/docs/Trainer/Distributed
@ -88,6 +88,9 @@ module avail
 module load cuda-10.0
 # ------------------------

+# make sure you've loaded the latest cuda version on your system
+module load gcc-9.1
+
 $ pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
 ```