当前位置:   article > 正文

报错:torch.distributed.elastic.multiprocessing.errors.ChildFailedError:

torch.distributed.elastic.multiprocessing.errors.childfailederror:

错误:

torch.distributed.elastic.multiprocessing.errors.ChildFailedError: 这个主要是torch的gpu版本和cuda不适配

我的nvcc -V是11.8

torch使用的:

pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118

但是我发现下这个也不行,就降低了一个小版本,但还是cu118 就OK了

pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118

Previous PyTorch Versions | PyTorch  附个地址,可以去寻找对应的gpu版本torch

!一定要cu版本和nvcc 的版本相对应!

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/从前慢现在也慢/article/detail/699968
推荐阅读
相关标签
  

闽ICP备14008679号