torch.cuda.is_available()返回false以及在pytorch官网下载遇到没有符合需求的版本的问题

一、测试pytorch CUDA是否可用

1、在cmd的Python环境下，或者在Anaconda Prompt下输入以下命令进行测试：

import torch
print(torch.__version__)
print(torch.cuda.is_available()) #cuda是否可用，返回为True表示可用
torch.cuda.device_count()#返回GPU的数量
torch.cuda.get_device_name(0)#返回gpu名字，设备索引默认从0开始

我的测试结果如下图：

torch.cuda.is_available()返回false，后来发现是因为没有使用pytorch官网，换源使用了清华源，下载的是CPU版本的torch。

二、 去pytorch官网下载

我下载的CUDA版本是CUDA10.2，去下载页面https://pytorch.org/get-started/locally/ 后发现不提供cuda10.2的pytorch下载，即图中最后一行所示内容。不过不要紧，点击下图圈中的内容可以下载之前的版本。

我是使用pip命令下载的，首先我尝试下载torch==1.11.0版本，即使用如下图提供的命令，将该命令复制到命令行窗口或者Anaconda Prompt窗口：

然后报错了，报错的信息如下：ERROR: Could not find a version that satisfies the requirement torch==1.11.0+cu102 (from versions: 1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.8.1+cu102, 1.9.0, 1.9.0+cu102, 1.9.1, 1.9.1+cu102, 1.10.0, 1.10.0+cu102, 1.10.1, 1.10.1+cu102, 1.10.2, 1.10.2+cu102, 1.11.0, 1.12.0, 1.12.1)
ERROR: No matching distribution found for torch==1.11.0+cu102。

意思就是，没有找到匹配的版本，然后括号中是有的版本。我点进那个命令：pip install torch==1.11.0+cu102 torchvision==0.12.0+cu102 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu102 后面的网址即：https://download.pytorch.org/whl/cu102 查看，发现确实没有符合Windows的这个版本。如下图所示：

然后，我回到可以安装pytorch之前版本的页面，鉴于之前提示1.10.0+cu102是有的，所以我往下拉找到了v1.10.0版本，如下图。

往下拉找到下载的 pip 命令，即划线处的：pip install torch==1.10.0+cu102 torchvision==0.11.0+cu102 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html。
然后运行此命令，如下，下载成功！有的地方说官网下载会很慢，我白天下载的时候没有遇到这种情况，速度还挺快的。

然后再次运行文章开头的测试pytorch CUDA是否可用的命令，结果如下：

torch.cuda.is_available()返回True，下载成功。