From cefcf4cd12c6b15fb592758787b322aed8781cb6 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 26 Sep 2019 15:27:34 -0400 Subject: [PATCH] Update Distributed training.md --- docs/Trainer/Distributed training.md | 3 +++ 1 file changed, 3 insertions(+) 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" ./ ```