Update Distributed training.md

2019-09-26 15:30:54 -04:00 · 2019-09-26 15:30:54 -04:00 · 059b2fae29
parent cefcf4cd12
commit 059b2fae29
1 changed files with 2 additions and 2 deletions
--- a/docs/Trainer/Distributed
+++ b/docs/Trainer/Distributed
@ -88,8 +88,8 @@ module avail
 module load cuda-10.0
 # ------------------------

-# make sure you've loaded the latest cuda version on your system
-module load gcc-9.1
+# make sure you've loaded a cuda version > 4.0 and < 7.0
+module load gcc-6.1.0

 $ pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
 ```