DOCKER里面PYTORCH关于GLOO地址声明 && NCCL地址声明

在Docker环境中进行多卡训练时遇到RuntimeError,无法找到my_username地址。解决方法是在训练前设置环境变量GLOO_SOCKET_IFNAME为lo,确保分布式训练时网络通信正常。若出现训练卡顿,可以设置NCCL_SOCKET_IFNAME=lo来优化通信性能。

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

      

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

CV-deeplearning

请博主加个火腿

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值