解决pytorch多进程ValueError: Error initializing torch.distributed using env:// rendezvou...报错

完整报错为:ValueError: Error initializing torch.distributed using env:// rendezvous: environment variable MASTER_ADDR expected, but not set

解决方法1:

在代码中使用:

import os

os.environ['MASTER_ADDR'] = 'localhost'
os.environ['MASTER_PORT'] = '5678'

解决方法2:

如果是运行命令行的,可以使用:

export MASTER_ADDR=localhost
export MASTER_PORT=5678

你可能感兴趣的:(pytorch/神经网络,pytorch)