解决torch.cuda.is_available()为False的问题

解决torch.cuda.is_available为False

问题：在Anaconda环境下，电脑拥有GPU并且已经通过conda install安装了Pytorch、cudatoolkit，但是torch.cuda.is_available()始终返回False（找不到显卡）

原因：使用conda install安装的pytorch、torchvision均为CPU版本，无法使用GPU进行计算。

解决：访问 镜像源 下载对应的GPU版本Pytorch、Torchvision
CPU版：

GPU版：

进入Anaconda的对应环境下，使用pip install 下载镜像路径名进行安装

完成后再次调用GPU

import torch
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

# Assuming that we are on a CUDA machine, this should print a CUDA device:

print(device)

成功返回gpu:0