PyTorch 2.9发布：增强AMD ROCm与英特尔 XPU安装支持

PyTorch 2.9于今日正式发布，适逢下周旧金山PyTorch大会开幕。本次版本最显著的特性是通过扩展wheel变体支持，显著提升了AMD ROCm与英特尔 XPU的安装体验。

基于PyTorch 2.8初期针对Windows平台英伟达 CUDA的Python wheel变体支持，2.9版本进一步将该功能延伸至Linux系统的AMD ROCm与英特尔 XPU平台。

wheel变体支持旨在优化硬件/软件平台检测能力，可自动识别平台属性并正确处理对应的Python软件包安装。这使得用户无需依赖不同软件包名称或指定wheel索引即可轻松安装PyTorch（待WheelNext标准广泛采用后还将惠及其他Python包），极大简化了特定计算平台的部署流程。

关于此项wheel变体支持如何助力AMD ROCm部署的更多技术背景，可参阅AMD官方博客。

PyTorch 2.9还通过支持OCP微缩放格式mx-fp8与mx-fp4，进一步强化了AMD ROCm的兼容性。该特性主要面向搭载ROCm 7.0的AMD GFX950硬件。

此外，新版本引入了对称内存功能以简化多GPU内核编程，为英特尔 GPU提供FlexAttention支持，实施ARM平台改进与优化，并包含一系列其他功能增强。

更多PyTorch 2.9更新详情请参阅GitHub发布说明与PyTorch官方公告。