diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 3ede697c76..6d4f639c01 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -336,7 +336,8 @@ class Trainer(TrainerIO): # set up server using proc 0's ip address # try to init for 20 times at max in case ports are taken - ip = self.__get_root_node_ip(self.proc_rank, self.nb_gpu_nodes) + # ip = self.__get_root_node_ip(self.proc_rank, self.nb_gpu_nodes) + ip = os.environ['MASTER_ADDR'] self.__init_tcp_connection(ip) # CHOOSE OPTIMIZER