IT之家 11 月 22 日消息,ROCm 是一個開源軟件平臺,允許研究人員利用 AMD Instinct 加速器的潛力,促進跨平臺的高性能計算和 AI 創(chuàng)新。
AMD 上個月宣布為基于 RDNA 3 的 Radeon Pro W7900 和 Radeon RX 7900 XTX 顯卡帶來了對 PyTorch 2.0.1 的 ROCm 支持,今天正式將其兼容范圍擴展到 RX 7900 XT 次旗艦游戲顯卡上。
AMD Radeon RX 7900 XT 基于 Navi 31 GPU 芯片,擁有 20 GB 超大顯存以及總共 168 個 AI 加速器單元,支持 ML 訓(xùn)練和推理加速。
據(jù)介紹,該 GPU 還支持基于 PyTorch 的 ML 模型和算法,這些模型和算法已經(jīng)專門針對 AMD 進行了優(yōu)化,并將與運行 22.04.3 版本(Ubuntu)的 Linux 操作系統(tǒng)兼容,感興趣的IT之家小伙伴可以前往官網(wǎng)下載。
這一舉措使開發(fā)人員能夠使用配備 AMD 消費級旗艦 GPU 利用 PyTorch 庫進行人工智能應(yīng)用的開發(fā),從而避免對于英偉達的過度依賴。
此外,該驅(qū)動專門針對 PyTorch 2.0.1 進行了優(yōu)化,適配了 AMD RDNA3-based GPU 的功能。
廣告聲明:文內(nèi)含有的對外跳轉(zhuǎn)鏈接(包括不限于超鏈接、二維碼、口令等形式),用于傳遞更多信息,節(jié)省甄選時間,結(jié)果僅供參考,IT之家所有文章均包含本聲明。