亲测有效解决torch.cuda.is_available()返回False的问题(分析+多种方案),点进不亏

文章目录

解决torch.cuda.is_available()返回False

出现返回False的原因

问题1:版本不匹配

电脑实际安装的CUDA版本 大于 已安装的pytorch所对应的CUDA的版本
注意:小于等于是可以的,但是不能大于

问题2:错下成了cpu版本的(小编正是这种问题)

可能使用了清华源镜像,但是清华源镜像中所对应的pytorch没有cuda版本的,这就造成了下载成了cpu版本的错误。

解决方案

方案一

针对于问题一,降低pytorch的版本 ,一般不建议去修改CUDA的版本,因为CUDA的版本会极有可能影响其他部分的使用。

方案二

针对于问题二,

  1. 打开 Anaconda Prompt ,输入 conda list,如下图所示

亲测有效解决torch.cuda.is_available()返回False的问题(分析+多种方案),点进不亏
2.卸载pytorch,输入 conda uninstall pytorch
亲测有效解决torch.cuda.is_available()返回False的问题(分析+多种方案),点进不亏

3.重新安装pytorch

官网链接:https://pytorch.org/

亲测有效解决torch.cuda.is_available()返回False的问题(分析+多种方案),点进不亏

在Anaconda Prompt,直接输入以下命令,注意不要用清华源(下面命令加了 -c pytorch,就意味着不再是使用清华源)

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

4.检验pytorch是否可用

亲测有效解决torch.cuda.is_available()返回False的问题(分析+多种方案),点进不亏

Original: https://blog.csdn.net/qq_44824148/article/details/121009580
Author: 零碎@流年絮语
Title: 亲测有效解决torch.cuda.is_available()返回False的问题(分析+多种方案),点进不亏

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/628341/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球