ROCm 是一个开源软件平台，允许研究人员利用 AMD Instinct 加速器的潜力，促进跨平台的高性能计算和 AI 创新。

AMD 本周发布了适用于 Ubuntu Linux 的 ROCm 5.7.1 驱动程序，为基于 RDNA 3 的 Radeon Pro W7900 和 Radeon RX 7900 XTX 显卡带来了对 PyTorch 2.0.1 的支持。

这一举措使开发人员能够使用配备 AMD 消费级旗舰 GPU 利用 PyTorch 库进行人工智能应用的开发，从而避免对于英伟达的过度依赖。

据介绍，该驱动专门针对 PyTorch 2.0.1 进行了优化，适配了 AMD RDNA3-based GPU 的功能。

不过目前 ROCm 5.7.1 驱动仅支持两款旗舰显卡：Radeon RX 7900 XTX 24GB 和 Radeon Pro W7900 48GB，因为像大型语言模型（LLM）这样的 AI 应用程序往往需要大容量显存，而 AMD 也希望尽可能多地销售高端硬件。

AMD 指出，Navi 31 GPU 内置了 192 个 AI 加速器，相比 Navi 21 的 96 个实现翻倍，可适用于人工智能应用。同时，AMD Radeon RX 7900 XTX 24 GB 定价 999 美元（国行首发价 7999 元，目前普遍低于该价），远低于专业 AI 和 HPC GPU。

AMD Radeon RX 7900 XTX 拥有 96CU，6144 流处理器，GPU 频率 1900-2500MHz，单精度性能可达 61TFLOPS。显存为 24GB 20Gbps GDDR6，带宽可达 960GB / s，加上 96MB 无限缓存，等效带宽可达 3.5TB / s。