RuntimeError: Address already in use pytorch分布式训练

RuntimeError: Address already in use

Pytorch 使用多GPU训练过程出现错误,具体错误如下:

出现该错误原因是通信端口被占用。

解决办法:

在train.py文件中加入:

os.environ['MASTER_PORT'] = '8888'  # 端口号随意指定

 之后重新训练即可正常

猜你喜欢

转载自blog.csdn.net/athrunsunny/article/details/129174827