3
1
AI人工智能
DCU torch版本是否支持nccl
2024-09-03 15:52:17
81

DCU的torch版本是否支持nccl呢,发现DistributedDataParallel使用nccl时socket timeout,但改为gloo则正常

 

全部评论(1)
只看楼主
牛
牛涛
1楼
你好,DCU torch是支持nccl的,麻烦将NCCL日志级别设置为info:NCCL_DEBUG=INFO,抓一下详细的日志。这里是一个resnet 50的网络,https://developer.hpccube.com/codes/modelzoo/resnet50-pytorch。可以使用这个网络测是一下,也可以用自己的网络。
2024/9/4
没有更多内容了
--
楼主
Level.1
0 帖子
楼主热帖
暂无数据