torch.cuda.is_available()为False

问题描述

anaconda创建了一个python3.8的环境，装了pytorch和cuda，但是cuda不能用，网上查询发现是国内镜像源的问题，国内镜像源下下来的torchvison等，可以在conda list查询结果的第三列发现带有“_cpu”标志。

解决办法

直接pip不行，失败原因忘记了，好像是网络相关。直接去下载whl文件，放在环境的“XXX(自定义环境名)\Lib\site-packages”下面，然后“pip install XXX.whl”, whl文件一共要下三个，分别是torch+cuda,torchvision,torchaudio, 版本要严格对应，不然你装了错误版本的torchvision又给你下载对应的新版本torch。版本的三个数字都要对应，只看1.8不行，要看1.8.1，下面是我的对应版本。

下载whl的网址

1.下载torch，cu111表示cuda，（我最后的cudatoolkit版本是11.3.1，打个问号）
链接: https://download.pytorch.org/whl/torch/.

2.下载torchvison, torchaudio等
链接: https://download.pytorch.org/whl/torch_stable.html.