你好,DCU torch是支持nccl的,麻烦将NCCL日志级别设置为info:NCCL_DEBUG=INFO,抓一下详细的日志。这里是一个resnet 50的网络,https://developer.hpccube.com/codes/modelzoo/resnet50-pytorch。可以使用这个网络测是一下,也可以用自己的网络。
DCU的torch版本是否支持nccl呢,发现DistributedDataParallel使用nccl时socket timeout,但改为gloo则正常

暂无数据