[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

安装完成后,再次尝试,成功!

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
乘风的头像乘风管理团队
上一篇 2023年8月17日
下一篇 2023年8月17日

相关推荐