華人大神出手AMD顯卡AI“煉丹”能力可達RTX 4090八成性能
除了玩遊戲,NVIDIA的顯卡還在AI加速上遙遙領先,不僅科技巨頭都在搶A100/H100等顯卡,GeForce遊戲卡也是普通人搞AI煉丹的選擇,AMD顯卡在這方面落後不少。好消息是AMD現在也能追上來了,深度學習領域的華人大神、OctoML CTO陳天奇帶領團隊在AMD顯卡上做出了大模型推理的方案,這意味著業界又有NVIDI之外的選擇,不用為N卡搶破頭、加價而困擾了。
根據他的優化算法,在Llama2的7B和13B大模型中,用RX 7900 XTX顯卡也能跑出讓人滿意的性能,性能已經達到了RTX 4090顯卡82%或者RTX 3090 Ti顯卡的94%水平。
不僅是性能追上來了,AMD顯卡的性價比也要高得多,RX 7900 XTX跟其他兩款顯卡都是24GB顯存,但只要999美元,RTX 4090、RTX 3090 Ti分別是1599、1999元,價格高出60-100%。
當然,RX 7900 XTX現在的價格早就鬆動了,實際售價比999美元便宜不少,而N卡價格一直堅挺。
究其根底,A卡在AI煉丹能力上的提升,尤其是RX 7900 XTX這樣的遊戲卡能有質變,還是AMD最近在ROCm軟件生態上的改進,以前後者對遊戲卡及Win平台的支持很差,4月份開始不僅支持Windows系統,也加入了對RX 7900及工作站顯卡的支持。
同時AMD還發布了ROCm HIP SDK,允許開發者將針對CUDA平台開發的代碼輕鬆遷移到AMD平台,這有是陳天奇等大神可以在AMD顯卡上推動AI能力提升的基礎。