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

        进入虚拟环境后pip list查看虚拟环境里的包,已有torch以及对应的版本号,但是torch调用cuda失败,情况如下:

[pytorch] torch.cuda.is_available() False 解决方法_第1张图片

 出现这种情况通常是下载了CPU版本的pytorch,一般通过命令下载的pytorch(包括官网提供的命令)是CPU版本的,可以通过查询anaconda里的安装包判断是否安装了CPU版本的pytorch。打开安装anaconda的磁盘/anaconda/pkgs文件:

[pytorch] torch.cuda.is_available() False 解决方法_第2张图片

        可以看到其实是安装的是CPU版本的pytorch,所以我们首先pip uninstall torch 将现在的CPU版本的pytorch卸载,顺便也把torchvision卸载,然后到下面的网站下载GPU版本的pytorch:

https://download.pytorch.org/whl/torch_stable.html

[pytorch] torch.cuda.is_available() False 解决方法_第3张图片

        进网站后,由于内容特别多,我们用搜索功能ctrl+f,比如我的cuda是11.0版本,就输入cu110,假如是cuda11.3版本,就输入cu113。然后选择想要的pytorch版本,cp38表示虚拟环境是python3.8的环境,torchvision也可以通过这个方法获得。 

        最后我们将下载的whl文件放到你所创建的虚拟环境的Script文件中

[pytorch] torch.cuda.is_available() False 解决方法_第4张图片

然后直接在该目录处输入cmd,进入命令行窗口

[pytorch] torch.cuda.is_available() False 解决方法_第5张图片

 输入:pip install 完整的whl文件名。比如pip install torch-1.7.1+cu110-cp38-cp38-win_amd64.whl

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

[pytorch] torch.cuda.is_available() False 解决方法_第6张图片

 

 

 

你可能感兴趣的:(学习内容与问题的解决方案等,pytorch,深度学习,python)