1年多以前，微軟宣佈和多家硬體廠商合作，在 Windows Subsystem for Linux（WSL）上對 GPU 加速的機器學習（ML）模型進行訓練。 微軟在 2020 年 6 月放出了預覽版。 今天，這個開源的 GitHub 專案已經退出預覽，正式開放提供。

微軟訓練了 TensorFlow-DirectML，這是 TensorFlow 的一個分叉，利用 DirectML 為在 Windows 10 和 WSL 上訓練 ML 模型提供跨廠商的硬體加速。 微軟表示，使用 TensorFlow-DirectML 相當容易，因為它可以通過運行” pip install tensorflow-directml” 命令在Python環境下安裝。 之後，它將自動與你現有的訓練模型的腳本集成。

微軟表示，它直接與學生和專業人士合作，為他們的腳本提供覆蓋，並優化批量歸一化和卷積等操作。 這個過程也涉及加強 GPU 調度和記憶體管理機制。 微軟與 Nvidia、AMD和英特爾等供應商合作，確保在 Windows 10 和 WSL 之間提供流暢的體驗，以便在支援 DirectX 12 的 GPU 上加速培訓。

就目前而言，TensorFlow-DirectML 的系統要求如下

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驅動

→ 英特爾HD Graphics 6xx或更新版本，以及28.20.100.8322 WSL驅動程式

→ NVIDIA GeForce GTX 9xx系列GPU或更新版本，以及460.20 WSL驅動

注意：目前不支援Python 3.8或更新版本。 要使用官方PyPi軟體包，需要使用CPython解釋器。