GIS之深度学习07:CUDNN教程(CUDA12.1配套)

时间:2024-04-14 22:17:53

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