确认CUDA和CUDNN安装成功方法

从现在开始，努力学习吧！本文《Windows确认CUDA和CUDNN安装成功方法》主要讲解了等等相关知识点，我会在golang学习网中持续更新相关的系列文章，欢迎大家关注并积极留言建议。下面就先一起来看一下本篇正文内容吧，希望能帮到你！

前言

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

进入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调用。

到这里，我们也就讲完了《确认CUDA和CUDNN安装成功方法》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于Windows,PyTorch,CUDA,cuDNN,安装确认的知识点！