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

js 原创
2023-10-23 电脑百科网

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,由于像大型语言模型(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。

小巧实用颜值高 2500元A10-9700迷你主机配置推荐 均衡高品质 AMD锐龙5 1500X搭GTX1060游戏配置推荐 IBM 云裸机服务器采用第三代 AMD EPYC:每台 128 核,支持 4TB 内存 绝地求生·大逃杀配置 5000元畅玩大逃杀电脑配置推荐 六核独显高性价比 3500左右FX-6330/RX460主流游戏配置推荐 美光为 AMD Radeon RX 6000 系列显卡推出 GDDR6 显存,带宽可达 512GB/s
热门文章
为你推荐