AMD 发布 ROCm 5.7，为 RX 7900 XTX 和 PRO W7900 显卡启用 PyTorch 支持

图片地址：https://img.ithome.com/newsuploadfiles/2023/10/ef3886c4-c9b0-4fa6-9f08-bcf11e4f5713.jpg?x-bce-process=image/format,f_auto

IT之家 10 月 22 日消息，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，因为像大型语言模型（