NVIDIA 为深度学习提供的 GPU 加速底层库。
它像汽修厂备好的专用套筒:卷积、激活这些重活都顺手趁手,模型一上工位就能拧得飞快。
常在底层给训练和推理提速,广泛藏在深度学习框架和视觉模型下面。
CUDA它建立在 CUDA 之上,负责把常见深度学习算子跑得更快。
PyTorch很多 PyTorch 的底层计算,会调用它来拿到 GPU 加速。
CNN卷积网络里的卷积和激活等操作,正是它最擅长优化的部分。
GPU它要靠 GPU 才能发挥价值,本质是在替 GPU 安排重活。