PyTorch 2.13 发布:支持苹果芯片并大幅优化显存
PyTorch 2.13 正式发布,带来多项关键性能优化与功能拓展。本次更新的核心亮点在于硬件与显存优化:FlexAttention 特性不再局限于 CUDA,正式扩展支持苹果芯片(Apple Silicon);同时针对大词表模型进行了底层优化,使其显存占用大幅降低达 4 倍,显著提升了相关模型的计算与运行效率。
2026-08-01 ~ 2026-08-01 · 2 条相关
- PyTorch 2.13 发布:FlexAttention 登陆苹果芯片,大词表模型显存暴降 4 倍 — PyTorch · 2026-08-01
另有 1 条近重复转述:PyTorch