当前位置:   article > 正文

PyTorch AttributeError: module ‘torch._C‘ has no attribute ‘_cuda_setDevice‘_attributeerror: module 'torch._c' has no attribute

attributeerror: module 'torch._c' has no attribute '_cuda_setdevice

这个问题说简单也很简单,大概率是你的pytorch安装了cpu版本...但现在默认安装下来就是cpu版本,可以在python中输入下面两行测试以下是不是不可用,如果显示False,那么就是不可用。

  1. import torch
  2. torch.cuda.is_available()

然后可以conda list看一下pytorch的版本,如果是cpu的版本,那么可以参考另一个写的很详细的博客。安装GPU版本的pytorch(解决pytorch安装时默认安装CPU版本的问题)保姆级教程_3EoIPrime的博客-CSDN博客

简述一下做法,就是使用conda删除onlycpu这个包,如果没有这个包,就先安装,再卸载。卸载的时候会有类似的提示。

  1. The following packages will be UPDATED:
  2. mkl anaconda/pkgs/main::mkl-2021.4.0-h06a~ --> anaconda/cloud/conda-forge::mkl-2021.4.0-h8d4b97c_729
  3. mkl_fft anaconda/pkgs/main::mkl_fft-1.3.1-py3~ --> anaconda/cloud/conda-forge::mkl_fft-1.3.1-py38h8666266_1
  4. The following packages will be SUPERSEDED by a higher-priority channel:
  5. pytorch pytorch::pytorch-2.0.0-py3.8_cpu_0 --> anaconda/cloud/pytorch::pytorch-1.7.0-py3.8_cuda11.0.221_cudnn8.0.3_0
  6. torchaudio pytorch::torchaudio-2.0.0-py38_cpu --> anaconda/cloud/pytorch::torchaudio-0.7.0-py38
  7. torchvision pytorch::torchvision-0.15.0-py38_cpu --> anaconda/cloud/pytorch::torchvision-0.8.1-py38_cu110
  8. The following packages will be DOWNGRADED:
  9. _openmp_mutex 4.5-2_gnu --> 4.5-2_kmp_llvm
  10. Proceed ([y]/n)? y

然后,就自动帮你把pytorch以及其它的包换成gpu的了。

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

闽ICP备14008679号