torch.cuda.is_available()false的可能原因

在cuda和cudann都已经安装并确定安装完成后,使用torch官网上提供的conda命令进行安装,但是安装完成后torch.cuda.is_available()依然返回false,查看了conda list发现安装的pytorch依然是cpu版
torch.cuda.is_available()false的可能原因_第1张图片
在网上搜寻一段时间后,我的解决方式如下:
在https://download.pytorch.org/whl/torch_stable.html上找到自己显卡适配的cuda版本,下载对应系统的torch,torchaudio和torchvision
torch.cuda.is_available()false的可能原因_第2张图片
如我下载了:
在这里插入图片描述
然后在终端下进入对应的目录,使用pip install filename 进行安装,三个分别安装完成后再进行torch.cuda.is_available()即返回true

你可能感兴趣的:(pytorch)