开源深度学习框架 PyTorch 2.9 正式发布:全面强化 AMD 与 Intel GPU 支持
开源深度学习框架 PyTorch 2.9 现已发布,本次更新重点强化了对 AMD ROCm 与 Intel XPU 平台的支持,让安装与部署更加便捷。
新版本引入了跨平台的 wheel 变体机制,可根据系统自动选择合适的硬件支持包,用户无需再手动配置安装参数。同时,PyTorch 2.9 还为 AMD ROCm 7.0 平台加入了 OCP 微缩格式(mx-fp8、mx-fp4) 支持,并新增了对称内存(Symmetric Memory)功能,使多 GPU 编程更加高效。
此外,Intel GPU 用户可获得 FlexAttention 注意力机制优化,ARM 平台也迎来多项性能提升。
整体来看,PyTorch 2.9 的发布标志着该框架在异构硬件生态上的进一步开放与成熟,为开发者在非 NVIDIA 平台上训练与部署 AI 模型提供了更完善的支持。
下载地址:https://github.com/pytorch/pytorch/releases/tag/v2.9.0
