cuDNN(常写作 CuDNN)是 NVIDIA 提供的 CUDA Deep Neural Network library,即用于 GPU 加速的深度神经网络基础库,包含卷积(convolution)、池化(pooling)、归一化(normalization)、RNN 等常用算子的高性能实现,常被 TensorFlow、PyTorch 等深度学习框架调用以提升训练与推理速度。
/ˌsiː juː ˌdiː ˌɛn ˈɛn/
I installed cuDNN to speed up my CNN training on the GPU.
我安装了 cuDNN 来加速在 GPU 上训练卷积神经网络。
Without the correct cuDNN version, the framework may fall back to slower CPU kernels or fail to load GPU operators.
如果 cuDNN 版本不匹配,框架可能会退回到更慢的 CPU 计算内核,甚至无法加载 GPU 算子。
cuDNN 来自 CUDA Deep Neural Network 的缩写:其中 CUDA 是 NVIDIA 的 GPU 计算平台与编程模型,DNN 指深度神经网络。该名称强调它是面向深度学习算子的 CUDA 加速库。