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

珠江路在线   2023年10月23日  【 转载 】风车动漫网 

  本文标签:AMD,radeon

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 美元(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 。

免责声明:凡标注转载/编译字样内容并非本站原创,转载目的在于传递更多信息,并不代表本网赞同其观点和对其真实性负责。