cuDNN(CUDA Deep Neural Network library):是NVIDIA打造的针对深度神经网络的加速库,是一个用于深层神经网络的GPU加速库。如果你要用GPU训练模型,cuDNN不是必须的,但是一般会采用这个加速库。windows系统下查看 CUDA 版本的命令如下:
在命令行中执行:
nvcc --version
或者进入 CUDA 的安装目录查看:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
查看 cuDNN 版本
进入 CUDA 的安装目录查看文件 cudnn.h :
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include\cudnn.h
如下所示,cuDNN 版本为 7.2.1 :