CUDNN(CUDA Deep Neural Network library)是NVIDIA专门针对深度学习应用开发的GPU加速库。它提供了一系列高效的深度学习算法的实现,包括卷积神经网络(CNN)、循环神经网络(RNN)等常用网络结构的基本操作,如卷积、池化、归一化、激活函数等。cuDNN针对NVIDIA GPU进行了高度优化,利用GPU的并行计算能力加速深度学习模型的训练和推理过程。通过使用cuDNN,开发者可以轻松地将深度学习模型部署到支持CUDA的NVIDIA GPU上,并获得显著的性能提升。
正文开始:
本文CUDA版本为12.1.1,所以CUDNN版本为12.x
需要登录NVDIA官网:Log in | NVIDIA Developer
登陆后选择版本
CUDA--11.x / CUDA--12.x
复制路径下的“bin” ;“include”;“lib”文件至CUDA路径下“
如图所示:
检查:
进入到CUDA的“demo_suite”路径下启动终端:
键入命令,显示“pass”即为成功
.\deviceQuery.exe
.\bandwidthTest.exe