NVIDIA 庆祝 PyTorch 月下载破 5000 万次,发布 CUDA 优化十年时间线

NVIDIAAI · x · 2026-10-20

NVIDIA 在 PyTorch Conference 2026 前发文庆祝与 PyTorch 的长期合作:支持 CUDA 的 PyTorch 月下载量已超 5000 万次。配图时间线梳理了从 2016 年 PyTorch 内置 CUDA 后端,到 FairScale FSDP、FP8 混合精度训练、PyTorch 2.0 用 Triton 生成 GPU kernel,再到 Blackwell 上的 FlexAttention 与 FlashAttention-4 支持等里程碑;另附系列博客,包括 MXFP8 训练 MoE 较 BF16 提速 1.3 倍、TorchTitan 上 DeepSeek-V3 预训练提速最多 41% 等成果。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →