1、cuDNN简介

cuDNN的全称为NVIDIA CUDA® Deep Neural Network library,是NVIDIA专门针对深度神经网络中的基础操作而设计并基于GPU的加速库。

cuDNN为深度神经网络中的标准流程提供了高度优化的实现方式,例如:convolution、pooling、normalization以及activation layers的前向以及后向过程。

安装CUDA的时候并没有安装cuDNN库,当开发者们需要用到深度学习GPU加速时才安装cuDNN库。

2、cuDNN与CUDA的版本对应关系

https://developer.nvidia.com/rdp/cudnn-archive

3、cuDNN下载地址

https://developer.nvidia.com/cudnn

4、cuDNN安装

下载好cuDNN的压缩包后进行解压,得到如下三个文件:

bin
include
lib

打开\NVIDIA Computing Toolkit\CUDA\v11.8,将cuDNN对应bin、lib、include三个文件与CUDA对应的bin、lib、include进行合并。

读者推荐

要想彻底的掌握pytorch的方方面面,推荐大家关注《PyTorch面试精华》