文章目录
- 解决torch.cuda.is_available()返回False
- 出现返回False的原因
- 问题1:版本不匹配
- 问题2:错下成了cpu版本的(小编正是这种问题)
- 解决方案
-
解决torch.cuda.is_available()返回False
出现返回False的原因
问题1:版本不匹配
电脑实际安装的CUDA版本 大于 已安装的pytorch所对应的CUDA的版本
注意:小于等于是可以的,但是不能大于
问题2:错下成了cpu版本的(小编正是这种问题)
可能使用了清华源镜像,但是清华源镜像中所对应的pytorch没有cuda版本的,这就造成了下载成了cpu版本的错误。
解决方案
方案一
针对于问题一,降低pytorch的版本 ,一般不建议去修改CUDA的版本,因为CUDA的版本会极有可能影响其他部分的使用。
方案二
针对于问题二,
- 打开 Anaconda Prompt ,输入
conda list
,如下图所示
![在这里插入图片描述](https://img-blog.csdnimg.cn/b39abacd1b2446a2a3802b44f050ce1f.png?x-oss-process=image/watermark,type_ZHJvaWRzYW5zZmFsbGJhY2s,shadow_50,text_Q1NETiBA6Zu256KOQOa1geW5tOe1ruivrQ==,size_20,color_FFFFFF,t_70,g_se,x_16)
2.卸载pytorch,输入conda uninstall pytorch
![在这里插入图片描述](https://img-blog.csdnimg.cn/3d87c44f4e1147e38f393f05d02d6430.png?x-oss-process=image/watermark,type_ZHJvaWRzYW5zZmFsbGJhY2s,shadow_50,text_Q1NETiBA6Zu256KOQOa1geW5tOe1ruivrQ==,size_20,color_FFFFFF,t_70,g_se,x_16)
3.重新安装pytorch
官网链接:https://pytorch.org/![在这里插入图片描述](https://img-blog.csdnimg.cn/8f27e2ca32de4f1da1233d87be11b2bd.png?x-oss-process=image/watermark,type_ZHJvaWRzYW5zZmFsbGJhY2s,shadow_50,text_Q1NETiBA6Zu256KOQOa1geW5tOe1ruivrQ==,size_20,color_FFFFFF,t_70,g_se,x_16)
在Anaconda Prompt,直接输入以下命令,注意不要用清华源(下面命令加了-c pytorch
,就意味着不再是使用清华源)
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
4.检验pytorch是否可用
![在这里插入图片描述](https://img-blog.csdnimg.cn/777d7042d53442e9960cff4b82300195.png?x-oss-process=image/watermark,type_ZHJvaWRzYW5zZmFsbGJhY2s,shadow_50,text_Q1NETiBA6Zu256KOQOa1geW5tOe1ruivrQ==,size_20,color_FFFFFF,t_70,g_se,x_16)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)