栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

解决问题:import torch失败和torch.cuda.is

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

解决问题:import torch失败和torch.cuda.is

安装过程比较曲折,阅读了很多dl的博客,并不是每一种都能顺利解决我自己的问题,但好在费尽周折之后安装成功了,于是将遇到的一些问题记录在这里,表述有问题的地方还请各位指出。

零、几点说明

1,本文默认已安装Anaconda,Win10系统
2,本人安装主要遇到的问题是:

  • conda list有pytorch但是import torch失败
  • import torch成功但是torch.cuda.is_available()返回false

3,主要参考的博客如下:
A, 两步搞定pytorch安装
B, Python安装Pytorch教程(图文详解)

一、Cuda和Cudnn未安装,安装后import torch成功

很多博客一般一开始都会让大家去查找Cuda的版本号,接着便放上官网的下载界面。但这样会让人以为,只要看到了版本号,就可以去下载pytorch了,但实际上不是这样,对于初次安装的人来说,可能他们连Cuda和Cudnn都还没有安装好,那么后面可能会出现import失败等问题。
以这篇比较火的博客为例,如何查看windows的CUDA版本,作者介绍了两种方法:

  • 去英伟达的控制面板里查找(这种方法简单直观,应该是大多数人的首选,==但是有问题!==详见下文)
  • 在命令行查询

第二种方法是作者在其他博客中经常看见的方法,于是记录了下来,但是他也坦陈,这种方法在他那里行不通。
事实上,第二种方法对我来说也行不通,遇到了一样的错误。

还是该博客,评论区里有位朋友很诚恳地指出了第一种方法的问题所在——“CUDA没有安装成功!!!

如果是这种情况,那么把相应的Cuda和Cudnn进行安装即可,这方面的博客有很多。
Cuda和cuDNN安装教程

二、用conda 还是用 pip?个人用conda安装成功

在解决了Cuda和Cudnn的安装问题后,发现,import torch成功但是torch.cuda.is_available()返回false
而根据这个主要参考的A博客:两步搞定pytorch安装
检查了一遍conda list,不幸被其言中,确实是下载的CPU版本,因为pytorch相关的行上有明显的cpu字样。

该博主信誓旦旦地说用** “必须手动去官网用pip下载安装” **,于是我信了,然后卸载conda uninstall pytorch,尝试用pip,但是不是速度慢就是找不到资源,于是又在网上疯狂找博客。
直到遇到B博客:Python安装Pytorch教程(图文详解)

高亮的部分,让我再次不幸被言中!


可能pip不适合我,于是还是回归了用conda安装。
同时这里的红字也给了我一个提醒,在官网选择pytorch对应的cuda版本时,可以选择比电脑上安装的cuda版本低一些!
于是我安装了11.3版本(之前11.6版本安装失败)

三、python版本合适否?

在看到这篇博客后,我在想我的python版本是不是也出了问题
Anaconda中安装pytorch后,验证import torch失败

此外,在解决第二个问题时,从博客上阅读得知,pip install是要安装特定版本的whl文件的

注意!stable版本是1.12.0,cuda版本暂时选择11.3

打开Run this Command后半部分的链接

点击torch,会看到有很多whl文件供选择。
因为stable版本是1.12.0,所以我只截取了1.12.0的部分
带有cpu字样的表明计算平台是cpu,显然不在选择范围内,操作系统字样是win的少之又少,而cp后面跟的数字是python的版本,36表示3.6, 37表示3.7.
此外,直观能看出,大部分都是python3.7版本往上了。

在查看conda list的时候,我发现pytorch这一行对应的字样是py3.7
而我的pytorch环境的python版本是3.6,为了保险起见我把该环境的python升级到了3.7。
然后重启了anaconda prompt

最后的界面如下,误打误撞也算是解决了!

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/1027014.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号