AMD為RX 7900 XTX提供ROCm和PyTorch支持,消費級產品加速AI/ML應用

近日,AMD 發佈 了適用於Ubuntu Linux的ROCm 5.7.1更新驅動程序,為基於RDNA 3架構的Radeon Pro W7900和Radeon RX 7900 XTX顯卡帶來了PyTorch 2.0.1的支持。此舉讓開發人員能夠使用PyTorch庫,將顯存容量分別為24GB和48GB的AMD旗艦級GPU用於人工智能(AI)和機器學習(ML)應用。

PyTorch是由Facebook AI Research開發的深度學習庫,作為一種開源工具,受到了業界的歡迎,並得到了CPU和GPU製造商的廣泛支持。AMD最初對整合PyTorch支持的反應較慢,特別是對消費類Radeon和Radeon PRO顯卡的支持更為遲緩,不過隨着ROCm 5.7.1的推測,顯然情況出現了較大的轉變。

目前僅支持旗艦顯卡也是考慮大型語言模型(LLM)這樣的人工智能應用需要大容量的顯存,同時還能讓AMD儘可能地銷售高端顯卡。雖然Radeon Pro W7900和Radeon RX 7900 XTX的定價都不低,不過比起數據中心使用的計算卡,那可是便宜太多了。通過專業領域的人工智能和機器學習支持擴展到消費級產品,可以促進專業與遊戲產品在功能上的融合,AMD也希望打造一個豐富的人工智能生態系統,並且讓開源平台壯大,為提高成本效益和擴大使用範圍鋪平道路。

今年4月就有 報道 稱,ROCm SDK即將登陸Windows操作系統,同時將擴展對消費級Radeon顯卡的支持。隨後AMD首席執行官蘇姿豐博士 確認 未來ROCm會在GPU支持方面會投入更多,加入更多的消費級Radeon顯卡,並致力於與社區合作為其用户提供更好的支持。