TensorFlow-DirectML開(kāi)放提供:在Windows 10/WSL上訓(xùn)練ML模型
1 年多以前,微軟宣布和多家硬件廠商合作,在 Windows Subsystem for Linux(WSL)上對(duì) GPU 加速的機(jī)器學(xué)習(xí)(ML)模型進(jìn)行訓(xùn)練。微軟在 2020 年 6 月放出了預(yù)覽版。今天,這個(gè)開(kāi)源的 GitHub 項(xiàng)目已經(jīng)退出預(yù)覽,正式開(kāi)放提供。
微軟訓(xùn)練了 TensorFlow-DirectML,這是 TensorFlow 的一個(gè)分叉,利用 DirectML 為在 Windows 10 和 WSL 上訓(xùn)練 ML 模型提供跨廠商的硬件加速。微軟表示,使用 TensorFlow-DirectML 相當(dāng)容易,因?yàn)樗梢酝ㄟ^(guò)運(yùn)行"pip install tensorflow-directml"命令在Python環(huán)境下安裝。之后,它將自動(dòng)與你現(xiàn)有的訓(xùn)練模型的腳本集成。
微軟表示,它直接與學(xué)生和專業(yè)人士合作,為他們的腳本提供覆蓋,并優(yōu)化批量歸一化和卷積等操作。這個(gè)過(guò)程也涉及加強(qiáng) GPU 調(diào)度和內(nèi)存管理機(jī)制。微軟與 Nvidia、AMD 和英特爾等供應(yīng)商合作,確保在 Windows 10 和 WSL 之間提供流暢的體驗(yàn),以便在支持 DirectX 12 的 GPU 上加速培訓(xùn)。
就目前而言,TensorFlow-DirectML 的系統(tǒng)要求如下
Windows 10
● 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 或更新版本
→ NVIDIA GeForce GTX 9xx 系列 GPU 或更新版本
注意:目前不支持 Python 3.8 或更新版本。要使用官方 PyPi 軟件包,需要使用 CPython 解釋器。NumPy 1.19.4 需要 KB4598291 才能在 Windows 上正常工作。
Windows Subsystem for Linux
● 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)程序
→ NVIDIA GeForce GTX 9xx系列GPU或更新版本,以及460.20 WSL驅(qū)動(dòng)
注意:目前不支持Python 3.8或更新版本。要使用官方PyPi軟件包,需要使用CPython解釋器。
































