Windows 11 和更新版本的 Windows 10 更新支援在適用於 Linux 的 Windows 子系統實例內執行現有的 ML 工具、連結庫和熱門架構,這些架構使用 NVIDIA CUDA 進行 GPU 硬體加速。 這包括 PyTorch 和 TensorFlow,以及原生 Linux 環境中可用的所有 Docker 和 NVIDIA Container Toolkit 支援。
安裝 Windows 11 或 Windows 10 版本 21H2
若要使用這些功能,您可以下載並安裝 Windows 11 或 Windows 10 版本 21H2。
安裝 GPU 驅動程式
下載並安裝 已啟用 NVIDIA CUDA 的 WSL 驅動程式 ,以搭配現有的 CUDA ML 工作流程使用。 如需要安裝哪些驅動程式的詳細資訊,請參閱:
安裝 WSL
安裝上述驅動程序之後,請確定您 啟用 WSL 並 安裝以 glibc 為基礎的發行版,例如 Ubuntu 或 Debian。 選取 [設定] 應用程式的 [Windows Update] 區段中的 [檢查更新],以確保您擁有最新的核心。
備註
請確保您已啟用「接收其他 Microsoft 產品更新」。 您可以在 [設定] 應用程式的 [Windows Update] 區段中的 [進階選項] 中找到它。
針對這些功能,您需要 5.10.43.3 或更高版本的核心版本。 您可以在 PowerShell 中執行下列命令來檢查版本號碼。
wsl cat /proc/version
開始使用 NVIDIA CUDA
現在,請遵循 WSL 使用者指南上的 NVIDIA CUDA 中的指示,您可以透過 NVIDIA Docker 開始使用 Linux 工作流程,或在 WSL 內安裝 PyTorch 或 TensorFlow 。
透過 WSL 上 CUDA 社群論壇分享 NVIDIA 支援的意見反應。