在最新版的AMD顯卡驅(qū)動(dòng)中,AMD聯(lián)手微軟大幅提升了RDNA2架構(gòu)顯卡的AI功能,其中RX 6900 XT、RX 6600 XT的AI推理可分別提升最多3.7倍和3.4倍。
不久前,微軟將TensorFlow帶到了Windows上,并發(fā)布了TensorFlow-DirectML。TensorFlow是谷歌推出的一個(gè)端到端開(kāi)源機(jī)器。
微軟通過(guò)將其與微軟DirectML機(jī)器API適配,讓用戶能夠在Windows 10以及其自帶的Linux子系統(tǒng)上使用GPU加速機(jī)器模型訓(xùn)練。
AMD的21.8.1版顯卡驅(qū)動(dòng)已經(jīng)集成正式版的TensorFlow-DirectML,對(duì)比預(yù)覽版的21.5.2驅(qū)動(dòng),RX 6900 XT、RX 6600 XT的AI推理可分別提升最多3.7倍、3.4倍。
AI Benchmark Alpha測(cè)試得分則可分別提升4.4倍、3.7倍。
微軟表示,GPU加速的TensorFlow-DirectML機(jī)器訓(xùn)練支持所有DX12兼容顯卡。
AMD RDNA2架構(gòu)的Radeon RX、Radeon PRO系列顯卡已經(jīng)率先完全支持,操作系統(tǒng)則支持Windows10、Windows 11。