已解决RuntimeError: Distributed package doesn‘t have NCCL built in

在运行Robust High-Resolution Video Matting with Temporal Guidance论文的代码过程中,遇到了如下问题:

RuntimeError: Distributed package doesn‘t have NCCL built in

windows不支持NCCL backend

所以需要将用到NCCL的地方替换

解决办法:
需要改两处:

在开头加上这两段:

import os

os.environ["PL_TORCH_DISTRIBUTED_BACKEND"] = "gloo"

将报错的地方改为gloo
在这里插入图片描述
运行成功

猜你喜欢

转载自blog.csdn.net/qq_52381360/article/details/129629424