diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 55fd27dd..23f426f6 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -406,20 +406,11 @@ class Trainer(TrainerIO): port = 12910 os.environ['MASTER_PORT'] = f'{port}' - def get_pids(port): - command = "sudo lsof -i :%s | awk '{print $2}'" % port - pids = subprocess.check_output(command, shell=True) - pids = pids.strip() - if pids: - pids = re.sub(' +', ' ', pids) - for pid in pids.split('\n'): - try: - yield int(pid) - except: - pass - - # get pids in this port - pids = set(get_pids(port)) + # check for pids + command = "lsof -i :%s | awk '{print $2}'" % port + pids = subprocess.check_output(command, shell=True) + pids = pids.strip() + pids = str(pids) # if no processes on this port, then we're good if len(pids) == 0: diff --git a/tests/debug.py b/tests/debug.py index 4e2675b5..9111c663 100644 --- a/tests/debug.py +++ b/tests/debug.py @@ -138,4 +138,5 @@ if __name__ == '__main__': command = "lsof -i :%s | awk '{print $2}'" % 12910 pids = subprocess.check_output(command, shell=True) pids = pids.strip() + print(pids)