From f6b98fe74f35795789f09395ac08e7d394701888 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sun, 21 Jul 2019 08:33:53 -0400 Subject: [PATCH] updated docs --- docs/Trainer/Distributed training.md | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/docs/Trainer/Distributed training.md b/docs/Trainer/Distributed training.md index 8c8e2fe0..09e8cd35 100644 --- a/docs/Trainer/Distributed training.md +++ b/docs/Trainer/Distributed training.md @@ -57,8 +57,10 @@ Make sure you're on a GPU machine. You can set as many GPUs as you want. In this setting, the model will run on all 8 GPUs at once using DataParallel under the hood. ```python # set these flags -os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" -os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2,3,4,5,6,7" +# lightning sets these flags for you automatically +# no need to set yourself +# os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" +# os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2,3,4,5,6,7" trainer = Trainer(gpus=[0,1,2,3,4,5,6,7])