• 首页 首页 icon
  • 工具库 工具库 icon
    • IP查询 IP查询 icon
  • 内容库 内容库 icon
    • 快讯库 快讯库 icon
    • 精品库 精品库 icon
    • 问答库 问答库 icon
  • 更多 更多 icon
    • 服务条款 服务条款 icon

[pytorch] torch.cuda.is_available() False 解决方法

武飞扬头像
没李不邢
帮助1

        进入虚拟环境后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
系列文章
更多 icon
同类精品
更多 icon
继续加载