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 美元(IT之家注:国行首发价 7999 元,目前普遍低于该价),远低于专业 AI 和 HPC GPU。
AMD Radeon RX 7900 XTX 拥有 96CU,6144 流处理器,GPU 频率 1900-2500MHz,单精度性能可达 61TFLOPS。显存为 24GB 20Gbps GDDR6,带宽可达 960GB / s,加上 96MB 无限缓存,等效带宽可达 3.5TB / s。
广告声明:文内含有的对外跳转链接(包括不限于超链接、二维码、口令等形式),用于传递更多信息,节省甄选时间,结果仅供参考,IT之家所有文章均包含本声明。
未经允许不得转载:新聚网 » AMD 发布 ROCm 5.7,为 RX 7900 XTX 和 PRO W7900 显卡启用 PyTorch 支持

新聚网
消息称 iQOO 轻薄电竞平板配备 8.35 英寸 LCD 屏,处于硬件定义阶段
上市价 3699 元 → 今日 1575 元:小米米家智能净烟机 P1 国补后再降新低
MONTECH 推出 NX400 系列风冷:单塔 4 热管直触,28mm 加厚风扇
上市价 3799 元:小米米家智能净烟机 P2 国补后 2295 元再降新低
48 通道 12 端口,井芯微发布首款全国产化 PCIe 4.0 交换芯片 JXW8848
华米 Active 2 Square 蓝宝石版手表开售:全天候健康监测、10 天常规续航,1099 元(国补价 934.15 元起)
全球最小、最亮商用投影仪:松下推出 PT-RQ45K,最高 4 万 ANSI 流明
黑鲨凤鸣挂耳式蓝牙耳机 Pro 银色款开售,首销价 279 元





