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 作为管理员,运行以下命令: