解决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版:
原创
2021-08-10 16:26:53 ·
2951 阅读 ·
0 评论