解决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