設置
  • 日夜間
    隨系統(tǒng)
    淺色
    深色
  • 主題色

AMD 發(fā)布 ROCm 5.7,為 RX 7900 XTX 和 PRO W7900 顯卡啟用 PyTorch 支持

2023/10/22 23:57:04 來源:IT之家 作者:問舟 責編:問舟

IT之家 10 月 22 日消息,ROCm 是一個開源軟件平臺,允許研究人員利用 AMD Instinct 加速器的潛力,促進跨平臺的高性能計算和 AI 創(chuàng)新。

AMD 本周發(fā)布了適用于 Ubuntu Linux 的 ROCm 5.7.1 驅(qū)動程序,為基于 RDNA 3 的 Radeon Pro W7900 和 Radeon RX 7900 XTX 顯卡帶來了對 PyTorch 2.0.1 的支持。

這一舉措使開發(fā)人員能夠使用配備 AMD 消費級旗艦 GPU 利用 PyTorch 庫進行人工智能應用的開發(fā),從而避免對于英偉達的過度依賴。

據(jù)介紹,該驅(qū)動專門針對 PyTorch 2.0.1 進行了優(yōu)化,適配了 AMD RDNA3-based GPU 的功能。

不過目前 ROCm 5.7.1 驅(qū)動僅支持兩款旗艦顯卡:Radeon RX 7900 XTX 24GB 和 Radeon Pro W7900 48GB,因為像大型語言模型(LLM)這樣的 AI 應用程序往往需要大容量顯存,而 AMD 也希望盡可能多地銷售高端硬件。

AMD 指出,Navi 31 GPU 內(nèi)置了 192 個 AI 加速器,相比 Navi 21 的 96 個實現(xiàn)翻倍,可適用于人工智能應用。同時,AMD Radeon RX 7900 XTX 24 GB 定價 999 美元(IT之家注:國行首發(fā)價 7999 元,目前普遍低于該價),遠低于專業(yè) AI 和 HPC GPU。

AMD Radeon RX 7900 XTX 擁有 96CU,6144 流處理器,GPU 頻率 1900-2500MHz,單精度性能可達 61TFLOPS。顯存為 24GB 20Gbps GDDR6,帶寬可達 960GB / s,加上 96MB 無限緩存,等效帶寬可達 3.5TB / s。

廣告聲明:文內(nèi)含有的對外跳轉(zhuǎn)鏈接(包括不限于超鏈接、二維碼、口令等形式),用于傳遞更多信息,節(jié)省甄選時間,結(jié)果僅供參考,IT之家所有文章均包含本聲明。

相關(guān)文章

關(guān)鍵詞:amd,顯卡

軟媒旗下網(wǎng)站: IT之家 最會買 - 返利返現(xiàn)優(yōu)惠券 iPhone之家 Win7之家 Win10之家 Win11之家

軟媒旗下軟件: 軟媒手機APP應用 魔方 最會買 要知