Advertisement

Pytorch 分布式dist.init_process_group报错NCCL 找不到GPU

阅读量:

完整报错信息如下:

复制代码
    Traceback (most recent call last):
      File "topFAR_COX_py2_fc1_pytorch.py", line 181, in <module>
    model = constructmodel(args)
      File "topFAR_COX_py2_fc1_pytorch.py", line 36, in constructmodel
    dist.init_process_group(backend='nccl', init_method=dist_url, rank=rank, world_size=world_size)
      File "/home/user1/miniconda3/lib/python3.7/site-packages/torch/distributed/distributed_c10d.py", line 510, in init_process_group
    timeout=timeout))
      File "/home/user1/miniconda3/lib/python3.7/site-packages/torch/distributed/distributed_c10d.py", line 603, in _ne

全部评论 (0)

还没有任何评论哟~