1. 问题描述
我在docker环境下,用多卡训练行人ReID时,报以下错误:
RuntimeError: [enforce fail at /pytorch/third_party/gloo/gloo/transport/tcp/device.cc:127] rp != nullptr. Unable to find address for: my_username
说找不到my_username地址。
2. 解决问题
在训练模型之前,先设置一下环境变量,即声明一下地址:
export GLOO_SOCKET_IFNAME=lo
3. 补充
如果多卡训练时出现卡顿,用下面的方法解决:
分布式训练前,终端先执行:export NCCL_SOCKET_IFNAME=lo
在Docker环境中进行多卡训练时遇到RuntimeError,无法找到my_username地址。解决方法是在训练前设置环境变量GLOO_SOCKET_IFNAME为lo,确保分布式训练时网络通信正常。若出现训练卡顿,可以设置NCCL_SOCKET_IFNAME=lo来优化通信性能。

1414

被折叠的 条评论
为什么被折叠?



