在微軟最新發(fā)布的 Windows Insider 預(yù)覽版本中，WSL2 獲得了 GPU 計(jì)算支持。這意味著Linux 二進(jìn)制文件可以利用 GPU 資源，在 WSL 中進(jìn)行機(jī)器學(xué)習(xí)、AI 開(kāi)發(fā)或是數(shù)據(jù)科學(xué)等工作。

微軟在今年五月份的 Build 2020 大會(huì)上宣布了 WSL 對(duì) GPU 計(jì)算的支持，對(duì)這項(xiàng)功能的需求在社區(qū)中一直擁有很高的呼聲。當(dāng)時(shí)，微軟表示，最初的 GPU 計(jì)算功能將提供兩種方案：

NVIDIA CUDA （CUDA 是跨平臺(tái)的 API，可以通過(guò) Windows 上的 WDDM GPU 抽象或 Linux 上的 NVIDIA GPU 抽象與 GPU 通信）

DirectML（直接機(jī)器學(xué)習(xí)，機(jī)器學(xué)習(xí)的低級(jí) API）

此次更新包括對(duì) NVIDIA CUDA 和 DirectML 的支持。CUDA 的初始預(yù)覽包括對(duì)現(xiàn)有機(jī)器學(xué)習(xí)工具、庫(kù)和流行框架（包括 PyTorch 和 TensorFlow）的支持，以及本機(jī) Linux 環(huán)境中可用的所有 Docker 和 NVIDIA Container Toolkit 支持，允許構(gòu)建在 Linux 上運(yùn)行的容器化 GPU 工作負(fù)載在 WSL 2 中按原樣運(yùn)行，讓開(kāi)發(fā)者可以在本地 Windows 計(jì)算機(jī)上進(jìn)行開(kāi)發(fā)和實(shí)驗(yàn)。它的面向目標(biāo)是專(zhuān)業(yè)人士。

DirectML 則更加針對(duì)初學(xué)者，能夠讓使用者跨 AMD、Intel 和 NVIDIA GPU，在 Windows 硬件上實(shí)現(xiàn)加速，體驗(yàn)機(jī)器學(xué)習(xí)的工作負(fù)載。這次預(yù)覽版本發(fā)布了帶有 DirectML 后端的 TensorFlow 預(yù)覽包，初學(xué)者可以從 TensorFlow 教程模型或微軟提供的示例開(kāi)始學(xué)習(xí)。

微軟表示，計(jì)劃在接下來(lái)的幾個(gè)月中開(kāi)源與 DirectML 一起使用的 TensorFlow 代碼擴(kuò)展庫(kù)。