，微軟最近幾天發(fā)布了TensorFlow的官方版本——DirectML，讓用戶可以在Windows Linux子系統(tǒng)上訓(xùn)練GPU加速機(jī)器學(xué)習(xí)模型。
顧名思義，TensorFlow—DirectML就是利用Google TensorFlow機(jī)器學(xué)習(xí)平臺適配微軟DirectML機(jī)器學(xué)習(xí)API，從而在Win10/WSL上訓(xùn)練機(jī)器學(xué)習(xí)模型時(shí)支持GPU加速。
IT之家了解到，用戶可以在Python環(huán)境下輸入pip install tensorflow—directml，支持在各種DirectX 12兼容硬件上進(jìn)行復(fù)雜機(jī)器學(xué)習(xí)模型的訓(xùn)練和推理。
Win10軟件和硬件要求:
Windows 10 1709，64位，
Python x86—64 3.5，3.6或3.7，
以下受支持的圖形處理器之一:
AMD R5/R7/R9 2xx系列或更高版本，
高清顯卡5xx或更高版本，
英偉達(dá)GeForce GTX 9xx系列GPU或更高版本，
WSL硬件和軟件要求:
Windows 10內(nèi)部預(yù)覽版，64位，
Python x86—64 3.5，3.6或3.7，
以下受支持的圖形處理器之一:
AMD R5/R7/R9 2xx系列或更高版本，以及20.20.01.05 WSL驅(qū)動(dòng)程序，
高清顯卡6xx或更高版本，以及28.20.100.8322 WSL驅(qū)動(dòng)程序，
英偉達(dá)GeForce GTX 9xx系列或更高版本，以及460.20 WSL驅(qū)動(dòng)程序，
具體用法可以查看TensorFlow—DirectML的GitHub頁面。
鄭重聲明：此文內(nèi)容為本網(wǎng)站轉(zhuǎn)載企業(yè)宣傳資訊，目的在于傳播更多信息，與本站立場無關(guān)。僅供讀者參考，并請自行核實(shí)相關(guān)內(nèi)容。
|
