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