RuntimeError: Distributed package doesn‘t have NCCL built in

项目场景:

提示:这里简述项目相关背景:

Improved Diffusion 复现

问题描述

提示:这里描述项目中遇到的问题:

RuntimeError: Distributed package doesn’t have NCCL built in

  File "D:\APP\Anaconda3\envs\diffusion\lib\site-packages\torch\distributed\distributed_c10d.py", line 602, in init_process_group
    default_pg = _new_process_group_helper(
  File "D:\APP\Anaconda3\envs\diffusion\lib\site-packages\torch\distributed\distributed_c10d.py", line 727, in _new_process_group_helper
    raise RuntimeError("Distributed package doesn't have NCCL " "built in")
RuntimeError: Distributed package doesn't have NCCL built in

原因分析:

提示:这里填写问题的分析:

windows不支持NCCL backend

解决方案:

提示:这里填写该问题的具体解决方案:

找到问题代码所在位置
D:\APP\Anaconda3\envs\diffusion\lib\site-packages\torch\distributed\distributed_c10d.py

    dist.init_process_group(backend=backend, init_method="env://")

更换为

    dist.init_process_group(backend="gloo", init_method="env://")

即可

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
心中带点小风骚的头像心中带点小风骚普通用户
上一篇 2023年7月12日
下一篇 2023年7月12日

相关推荐