torch.cuda.is_available()总是报false

显卡没问题,cuda和cuDNN版本也没有问题,整个流程都是按别人成功过的步骤来的,但最后总是报错。

我的各个版本如下:

python:3.7

cuda:10.0

torch.cuda.is_available()总是报false_第1张图片

解决办法:

开始跟着教程用conda命令行安装的pytorch,最后改用pip安装whl后成功了,也没搞明白是什么原因。

从官网上下载对应版本的whl以后,再pip:

pip install torch-1.6.0+cu101-cp37-cp37m-win_amd64.whl

安装好以后就true了。

 torch.cuda.is_available()总是报false_第2张图片

你可能感兴趣的:(gpu,cuda)