Windows 上使用 Docker GPU需要進行一系列的配置和步驟。這是因為 Docker 在 Windows 上的運行環境與 Linux 有所不同,需要借助 WSL 2(Windows Subsystem for Linux 2)和 NVIDIA Container Toolkit 來實現 GPU 的支持。以下是詳細的流程:
一、環境準備
1.系統要求
- Windows 10 或 Windows Server 2019/2022:64 位版本。
- WSL 2(Windows Subsystem for Linux 2):NVIDIA Container Toolkit 主要在 Linux 環境中運行,因此需要通過 WSL 2 使用。
- NVIDIA GPU 驅動:確保已安裝支持 CUDA 的 NVIDIA 顯卡驅動。
- Docker Desktop:需安裝并配置 Docker Desktop,并啟用 WSL 2 后端。
2.安裝 Docker Desktop
- 訪問 Docker 官網下載適用于 Windows 的 Docker Desktop。
- 雙擊安裝程序,按照提示完成安裝。
- 安裝完成后,啟動 Docker Desktop。
3.啟用 WSL 2 和虛擬機平臺
- 打開 PowerShell 作為管理員,運行以下命令: