NVIDIA cuDNN是用于深度神经网络的GPU加速库,它可以集成到TensorFlow的机器学习框架中。简单的插入式设计可以让开发人员专注于设计和实现神经网络模型,而不是简单调整性能,同时还可以在GPU上实现高性能现代并行计算。
选择对应CUDA的cuDNN版本
https://developer.nvidia.com/rdp/cudnn-archive
3. 下载完成后,进行解压缩,并将解压后文件中的cuda文件夹复制到CUDA安装目录下
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6
cuDNN中bin目录下的文件移动到 CUDA 的 bin 目录中
cuDNN目录中的 include 中的文件移动到 CUDA 的 include 目录中
cuDNN目录中的 lib 中的文件移动到 CUDA 的 lib 目录中
安装检测
cmd打开
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6\extras\demo_suite
分别运行:bandwidthTest.exe 和deviceQuery.exe
如果能看到结果就完成了cuDNN的安装。