在cuda和cudann都已经安装并确定安装完成后,使用torch官网上提供的conda命令进行安装,但是安装完成后torch.cuda.is_available()依然返回false,查看了conda list发现安装的pytorch依然是cpu版

在网上搜寻一段时间后,我的解决方式如下:
在https://download.pytorch/whl/torch_stable.html上找到自己显卡适配的cuda版本,下载对应系统的torch,torchaudio和torchvision

如我下载了:

然后在终端下进入对应的目录,使用pip install filename 进行安装,三个分别安装完成后再进行torch.cuda.is_available()即返回true

更多推荐

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