开源深度学习框架 PyTorch 2.9 正式发布:全面强化 AMD 与 Intel GPU 支持

开源深度学习框架 pytorch 2.9 正式上线,此次版本更新着重增强了对 amd rocm 及 intel xpu 架构的支持,显著提升了安装与部署的便利性。

开源深度学习框架 PyTorch 2.9 正式发布:全面强化 AMD 与 Intel GPU 支持

开源深度学习框架 PyTorch 2.9 正式发布:全面强化 AMD 与 Intel GPU 支持

本版本引入了全新的跨平台 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://www.php.cn/link/c8c1efaca9730df9826ea5f5bd72d025

源码地址:点击下载

网友留言(0 条)

发表评论