AI Rookies

CUDA — 统一计算设备架构

事实

NVIDIA 提供的 GPU 通用并行计算平台与编程接口。

人话

CUDA 像给显卡发了工牌:本来在网吧打游戏,现在进机房拧螺丝、搬大件、连夜赶工。

它让程序把计算丢给 GPU 并行处理,是训练和推理加速的底层通道。

相关概念

GPU
CUDA 把 GPU 从图形芯片变成可编程计算设备。

Pytorch
很多深度学习框架通过 CUDA 调用显卡算力。

Inference engine
推理引擎常依赖 CUDA 执行底层高性能计算。

Flash Attention
这类高性能算子通常需要在 CUDA 上实现。