[pytorch] torch.cuda.is_available() False 解决方法
进入虚拟环境后pip list查看虚拟环境里的包,已有torch以及对应的版本号,但是torch调用cuda失败,情况如下:
出现这种情况通常是下载了CPU版本的pytorch,一般通过命令下载的pytorch(包括官网提供的命令)是CPU版本的,可以通过查询anaconda里的安装包判断是否安装了CPU版本的pytorch。打开安装anaconda的磁盘/anaconda/pkgs文件:
可以看到其实是安装的是CPU版本的pytorch,所以我们首先pip uninstall torch 将现在的CPU版本的pytorch卸载,顺便也把torchvision卸载。
方法一:进入下面这个网址找到对应cuda版本的pytorch,然后复制pip下载。
Previous PyTorch Versions | PyTorch
方法二:然后到下面的网站下载GPU版本的pytorch:
Previous PyTorch Versions | PyTorch
进网站后,由于内容特别多,我们用搜索功能ctrl f,比如我的cuda是11.0版本,就输入cu110,假如是cuda11.3版本,就输入cu113。然后选择想要的pytorch版本,cp38表示虚拟环境是python3.8的环境,torchvision也可以通过这个方法获得。
最后我们将下载的whl文件放到你所创建的虚拟环境的Script文件中
然后直接在该目录处输入cmd,进入命令行窗口
输入:pip install 完整的whl文件名。比如pip install torch-1.7.1 cu110-cp38-cp38-win_amd64.whl
安装完成后,再次尝试,成功!
这篇好文章是转载于:学新通技术网
- 版权申明: 本站部分内容来自互联网,仅供学习及演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,请提供相关证据及您的身份证明,我们将在收到邮件后48小时内删除。
- 本站站名: 学新通技术网
- 本文地址: /boutique/detail/tanhiagkak
-
photoshop保存的图片太大微信发不了怎么办
PHP中文网 06-15 -
word里面弄一个表格后上面的标题会跑到下面怎么办
PHP中文网 06-20 -
photoshop扩展功能面板显示灰色怎么办
PHP中文网 06-14 -
《学习通》视频自动暂停处理方法
HelloWorld317 07-05 -
Android 11 保存文件到外部存储,并分享文件
Luke 10-12 -
TikTok加速器哪个好免费的TK加速器推荐
TK小达人 10-01 -
微信公众号没有声音提示怎么办
PHP中文网 03-31 -
excel下划线不显示怎么办
PHP中文网 06-23 -
excel打印预览压线压字怎么办
PHP中文网 06-22 -
微信运动停用后别人还能看到步数吗
PHP中文网 07-22