From dc87a4fc9168e92858497cd87ee98d1eb4917d55 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Fri, 12 Jul 2019 15:59:33 -0400 Subject: [PATCH] fixed nccl init --- pytorch_lightning/models/trainer.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index d84ea554..1606d99a 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -379,9 +379,6 @@ class Trainer(TrainerIO): :param tries: :return: """ - # hack to get nccl to stop throwing error... seems to be an nccl race condition - if self.proc_rank > 0: - sleep(10.0) root_node = os.environ['SLURM_NODELIST'].split(' ')[0] os.environ['MASTER_ADDR'] = root_node