在微軟最新發(fā)布的 Windows Insider 預(yù)覽版本中，WSL2 獲得了 GPU 計算支持。這意味著Linux 二進制文件可以利用 GPU 資源，在 WSL 中進行機器學(xué)習(xí)、AI 開發(fā)或是數(shù)據(jù)科學(xué)等工作。
微軟在今年五月份的 Build 2020 大會上宣布了 WSL 對 GPU 計算的支持，對這項功能的需求在社區(qū)中一直擁有很高的呼聲。當(dāng)時，微軟表示，最初的 GPU 計算功能將提供兩種方案：
NVIDIA CUDA （CUDA 是跨平臺的 API，可以通過 Windows 上的 WDDM GPU 抽象或 Linux 上的 NVIDIA GPU 抽象與 GPU 通信）
DirectML（直接機器學(xué)習(xí)，機器學(xué)習(xí)的低級 API）
此次更新包括對 NVIDIA CUDA 和 DirectML 的支持。CUDA 的初始預(yù)覽包括對現(xiàn)有機器學(xué)習(xí)工具、庫和流行框架（包括 PyTorch 和 TensorFlow）的支持，以及本機 Linux 環(huán)境中可用的所有 Docker 和 NVIDIA Container Toolkit 支持，允許構(gòu)建在 Linux 上運行的容器化 GPU 工作負載在 WSL 2 中按原樣運行，讓開發(fā)者可以在本地 Windows 計算機上進行開發(fā)和實驗。它的面向目標是專業(yè)人士。
DirectML 則更加針對初學(xué)者，能夠讓使用者跨 AMD、Intel 和 NVIDIA GPU，在 Windows 硬件上實現(xiàn)加速，體驗機器學(xué)習(xí)的工作負載。這次預(yù)覽版本發(fā)布了帶有 DirectML 后端的 TensorFlow 預(yù)覽包，初學(xué)者可以從 TensorFlow 教程模型或微軟提供的示例開始學(xué)習(xí)。
微軟表示，計劃在接下來的幾個月中開源與 DirectML 一起使用的 TensorFlow 代碼擴展庫。
下載文檔
