From 08e1ab64b528a15c7c165c946989c94dde7a4b87 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Fri, 12 Jul 2019 15:53:45 -0400
Subject: [PATCH] fixed nccl init

---
 pytorch_lightning/models/trainer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py
index 725ac147..64fbf16b 100644
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@@ -375,6 +375,7 @@ class Trainer(TrainerIO):
         :param tries:
         :return:
         """
+        sleep(self.proc_rank*2)
         root_node = os.environ['SLURM_NODELIST'].split(' ')[0]
         os.environ['MASTER_ADDR'] = root_node
         os.environ['MASTER_PORT'] = f'{port}'