diff --git a/docs/Trainer/Distributed training.md b/docs/Trainer/Distributed training.md index 0e7305ce0a..a5fdb24dd9 100644 --- a/docs/Trainer/Distributed training.md +++ b/docs/Trainer/Distributed training.md @@ -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" ./ ```