PyTorch 2.11 Released
2 months ago
- #PyTorch
- #GPU Optimization
- #Machine Learning
- PyTorch 2.11版本包含来自432位贡献者的2723次提交
- 核心特性:分布式训练的可微分集合操作、搭载FlashAttention-4后端的FlexAttention、MPS算子扩展、RNN/LSTM GPU导出支持、面向英特尔GPU的XPU Graph
- MPS改进包括错误报告和算子覆盖范围扩展
- RNN/LSTM模块现支持GPU导出,兼容动态形状
- ROCm设备端断言和TopK优化提升AMD GPU性能
- XPUGraph降低英特尔GPU工作负载的CPU开销
- 通过OpenBLAS实现FP16半精度GEMM运算,支持CPU部署场景
- CUDA 13现为默认版本;Torchscript已弃用,推荐转向torch.export
- 自2026年起发布周期将缩短为每2个月一次