PyTorch 2.9 发布，支持 AMD ROCm 和 Intel XPU 的安装更加便捷-Linuxeden开源社区

PyTorch 2.9 今日发布，下周将在旧金山举办的 PyTorch 会议之前。PyTorch 2.9 的亮点是更好的 AMD ROCm 和 Intel XPU 安装支持，通过扩展 wheel 变体支持实现。

在 PyTorch 2.8 中，Python wheel 变体支持最初是为了满足 Windows 上的 NVIDIA CUDA 而提供的，现在 Python 2.9 将 wheel 变体支持扩展到了 Linux 上的 AMD ROCm 和 Intel XPU 平台。

轮变体支持是为了更好地检测硬件/软件平台，以便自动检测平台属性，正确处理正确的 Python 包安装。这使得在 WheelNext 标准被广泛采用后，安装 PyTorch（以及其他 Python 包）变得更加容易，无需使用不同的包名或指定轮索引来获取适合您给定硬件/软件计算平台的正确包。

有关 PyTorch 中轮变体支持的相关背景信息，可以通过这篇 AMD 博客文章找到。

PyTorch 2.9 还增强了 AMD ROCm 支持，支持 mx-fp8 和 mx-fp4 的 OCP 微缩格式。这主要针对 AMD GFX950 硬件和 ROCm 7.0。

PyTorch 2.9 发布，支持 AMD ROCm 和 Intel XPU 的安装更加便捷

PyTorch 2.9 还引入了对多 GPU 内核编程的对称内存支持，FlexAttention 对 Intel GPU 的支持，ARM 平台改进和优化，以及其他各种增强功能。

更多关于 PyTorch 2.9 众多变更的详细信息，请通过 GitHub 和 PyTorch.org 公告获取。