检测代码：
import torch
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(torch.cuda.is_available())
用这两条条代码可以检测是否能找到GPU，如果找不到可能有如下几个原因：
1.没有安装CUDA/CUDNN
安装CUDA网址：https://developer.nvidia.cn/cuda-downloads
安装CUDNN：https://developer.nvidia.cn/rdp/cudnn-archive
记得CUDA要配准环境
注意1：如果直接搜索CUDA/CUDNN网址打不开是因为使用了.com的网址，用.cn就可以正常访问了
注意2：CUDA和CUDNN的版本一定要对应好，截止2022年3月建议安装11.3之前的版本，因为最新的版本可能跟pytorch还没有适配
2. CUDA和pytorch的版本无法匹配
这条原因最常见，CUDA和pytorch的对应版本怎么查找呢？
到这个网址：https://download.pytorch.org/whl/torch_stable.html
这里的CU113指的是CUDA11.3版本，torch-1.10.1指的是torch版本，cp-36指的是python 3.6版本，linux，win这些就不用我解释了吧。
那whl文件的使用呢？
我们将它下载到桌面，用WIN+R打开cmd，然后进入到桌面，利用pip安装即可
Original: https://blog.csdn.net/weixin_44952817/article/details/123294673
Author: 薛猫颚的腚
Title: 解决pytorch检测不到cuda的问题/pytorch找不到GPU
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/625310/
转载文章受原作者版权保护。转载请注明原作者出处！