IT之家 9 月 12 日消息 微軟近日發(fā)布了 TensorFlow-DirectML 的正式版，該工具可讓用戶(hù)在 Windows 的 Linux 子系統(tǒng)（WSL）上 GPU 加速機(jī)器學(xué)習(xí)（ML）模型訓(xùn)練。
顧名思義，TensorFlow-DirectML 用于將谷歌 TensorFlow 機(jī)器學(xué)習(xí)平臺(tái)與微軟 DirectML 機(jī)器學(xué)習(xí) API 適配，以支持在 Win10/WSL 上進(jìn)行機(jī)器學(xué)習(xí)模型訓(xùn)練時(shí)獲得 GPU 加速。
IT之家了解到，用戶(hù)可以在 Python 環(huán)境下（Python 3.8 及以上暫不支持）輸入 pip install tensorflow-directml 進(jìn)行安裝，支持在各種 DirectX 12 兼容硬件上訓(xùn)練和推理復(fù)雜的機(jī)器學(xué)習(xí)模型。
Win10 軟硬件要求：
Windows 10 1709，64 位（Build 16299 或更高）；
Python x86-64 3.5、3.6 或 3.7；
以下支持的 GPU 之一：
AMD Radeon R5/R7/R9 2xx 系列或更新版本；
英特爾 HD Graphics 5xx 或更新版本；
英偉達(dá) GeForce GTX 9xx 系列 GPU 或更新版本；
WSL 軟硬件要求：
Windows 10 Insider Preview，64 位（Build 20150 或更高）；
Python x86-64 3.5、3.6 或 3.7；
以下支持的 GPU 之一：
AMD Radeon R5/R7/R9 2xx 系列或更新版本，以及 20.20.01.05 WSL 驅(qū)動(dòng)；
英特爾 HD Graphics 6xx 或更新版本，以及 28.20.100.8322 WSL 驅(qū)動(dòng)；
英偉達(dá) GeForce GTX 9xx 系列或更新版本，以及 460.20 WSL 驅(qū)動(dòng)；
具體使用方法，大家可以查看 TensorFlow-DirectML 的 GitHub 頁(yè)面。
廣告聲明：文內(nèi)含有的對(duì)外跳轉(zhuǎn)鏈接（包括不限于超鏈接、二維碼、口令等形式），用于傳遞更多信息，節(jié)省甄選時(shí)間，結(jié)果僅供參考，IT之家所有文章均包含本聲明。