Windows确认CUDA和CUDNN安装成功方法

前言

如果你觉得这篇文章对你有帮助，别忘了点赞、收藏或留下评论，这将是作者持续创作的最大动力和支持。

进入CUDA安装目录

打开cmd命令行工具，将deviceQuery.exe文件拖入窗口，按下回车执行

若出现类似结果，表示第一项设备检测已通过

接着，同样将bandwidthTest.exe拖入命令行并回车运行

随后依次在Python环境中运行以下代码：

>>> import torch
>>> print(torch.backends.cudnn.version())
# 正常输出应为8200
>>> from torch.backends import cudnn  # 若无报错则说明导入成功
>>> cudnn.is_available()  # 正常情况下返回True
>>> a = torch.tensor(1.)
>>> cudnn.is_acceptable(a.cuda())  # 正常应返回True

如图所示输出结果，表明CUDNN已正确安装并可被PyTorch调用。