PyTorch AttributeError: module ‘torch._C‘ has no attribute ‘_cuda

这篇具有很好参考价值的文章主要介绍了PyTorch AttributeError: module ‘torch._C‘ has no attribute ‘_cuda_setDevice‘。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

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

import torch
torch.cuda.is_available()

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

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

The following packages will be UPDATED:

  mkl                anaconda/pkgs/main::mkl-2021.4.0-h06a~ --> anaconda/cloud/conda-forge::mkl-2021.4.0-h8d4b97c_729
  mkl_fft            anaconda/pkgs/main::mkl_fft-1.3.1-py3~ --> anaconda/cloud/conda-forge::mkl_fft-1.3.1-py38h8666266_1

The following packages will be SUPERSEDED by a higher-priority channel:

  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
  torchaudio             pytorch::torchaudio-2.0.0-py38_cpu --> anaconda/cloud/pytorch::torchaudio-0.7.0-py38
  torchvision          pytorch::torchvision-0.15.0-py38_cpu --> anaconda/cloud/pytorch::torchvision-0.8.1-py38_cu110

The following packages will be DOWNGRADED:

  _openmp_mutex                                   4.5-2_gnu --> 4.5-2_kmp_llvm


Proceed ([y]/n)? y

然后，就自动帮你把pytorch以及其它的包换成gpu的了。文章来源地址https://www.toymoban.com/news/detail-636174.html

到了这里，关于PyTorch AttributeError: module ‘torch._C‘ has no attribute ‘_cuda_setDevice‘的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！