近日，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顯卡，並致力於與社區合作為其用戶提供更好的支持。
來源：超能網
