AI Rookies

CuDNN — CUDA 深度神经网络库

事实

NVIDIA 为深度学习提供的 GPU 加速底层库。

人话

它像汽修厂备好的专用套筒:卷积、激活这些重活都顺手趁手,模型一上工位就能拧得飞快。

常在底层给训练和推理提速,广泛藏在深度学习框架和视觉模型下面。

相关概念

CUDA
它建立在 CUDA 之上,负责把常见深度学习算子跑得更快。

PyTorch
很多 PyTorch 的底层计算,会调用它来拿到 GPU 加速。

CNN
卷积网络里的卷积和激活等操作,正是它最擅长优化的部分。

GPU
它要靠 GPU 才能发挥价值,本质是在替 GPU 安排重活。