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