AMD 推出了 适用于 Ubuntu Linux 的更新版 ROCm 5.7.1 驱动程序，为基于 RDNA 3 的 Radeon Pro W7900 和 Radeon RX 7900 XTX 显卡带来了 PyTorch 2.0.1 支持。此举使开发人员能够将具有 24GB 和 48GB 内存的 AMD 旗舰客户端 GPU 用于使用 PyTorch 库的人工智能应用程序。

“我们很高兴能够使用基于 AMD Radeon RX 7900 XTX 和 Radeon Pro W7900 GPU 以及 ROCm 开放软件平台构建的 PyTorch 为 AI 社区提供机器学习开发新支持。这是我们第一个基于 RDNA 3 架构的实现，我们AMD 的 Radeon 产品管理副总裁 Dan Wood 表示：“我们期待与社区合作。”

该驱动程序更新专门针对流行的深度学习库 PyTorch 2.0.1 进行了调整，带来了 AMD 基于 RDNA3 的 GPU 的功能。ROCm 5.7.1 驱动程序仅支持旗舰显卡：Radeon RX 7900 XTX 24GB 和 Radeon Pro W7900 48GB。这是合乎逻辑的，因为大型语言模型 (LLM) 等人工智能应用程序受益于大容量板载内存，而且 AMD 希望销售尽可能多的高端主板。

AMD 指出，其 Navi 31 图形处理器拥有 192 个内置 AI 加速器，高于 Navi 21 的 96 个，使其成为适用于人工智能应用的可行 GPU。与此同时，AMD 的 Radeon RX 7900 XTX 24 GB 是一款消费级显卡，建议零售价为 999 美元，只是专业 AI 和 HPC GPU 成本的一小部分。

AMD 通过将传统上为专业领域保留的复杂机器学习支持扩展到消费级产品，促进了游戏和专业 GPU 功能的微妙融合。AMD 希望培育一个丰富的生态系统，让开源平台蓬勃发展，为提高成本效率和更广泛的可访问性铺平道路。