From 97b6ebccc05644921c938b25690ac0d9f482eeb4 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 26 Sep 2019 09:36:03 -0400 Subject: [PATCH] expanded apex install (#255) --- docs/Trainer/Distributed training.md | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/docs/Trainer/Distributed training.md b/docs/Trainer/Distributed training.md index c895e2472d..d1c52bd93e 100644 --- a/docs/Trainer/Distributed training.md +++ b/docs/Trainer/Distributed training.md @@ -74,6 +74,18 @@ First, install apex (if install fails, look [here](https://github.com/NVIDIA/ape ```bash $ git clone https://github.com/NVIDIA/apex $ cd apex + +# ------------------------ +# OPTIONAL: on your cluster you might need to load cuda 10 or 9 +# depending on how you installed PyTorch + +# see available modules +module avail + +# load correct cuda before install +module load cuda-10.0 +# ------------------------ + $ pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./ ```