windows docker cuda
时间: 2025-02-19 13:35:37 浏览: 64
### 如何在 Windows 上设置 Docker 以启用 CUDA 支持
#### 安装 NVIDIA 驱动程序
为了使 Docker 能够利用 GPU 加速功能,在 Windows 上必须先安装适用于 Windows 的最新版本的 NVIDIA 显卡驱动程序[^1]。
#### 安装 Docker Desktop
确保已下载并安装了最新的 Docker Desktop 版本。Docker Desktop 是官方推荐用于 Windows 平台上的 Docker 运行环境,它包含了必要的组件来支持 Linux 和 Windows 容器,并且集成了对 WSL 2 (Windows Subsystem for Linux 2) 的良好支持。
#### 启用 WSL 2 及其集成特性
由于目前大多数深度学习框架都是基于 Linux 构建的,因此建议开启 WSL 2 来作为默认的 Linux 子系统。这可以通过 PowerShell 使用管理员权限执行命令 `wsl --set-default-version 2` 来完成。接着,通过 Docker Desktop 设置界面中的选项激活与 WSL 2 的互操作性。
#### 安装 NVIDIA Container Toolkit on Docker Desktop
要让 Docker 容器能够访问主机系统的 GPU 设备,则需按照 NVIDIA 提供的方法安装特定于 Windows 的 NVIDIA Container Toolkit 插件。此插件允许用户轻松创建具有 GPU 访问权的新容器实例或更新现有镜像以便它们可以识别和使用 GPU 硬件资源。
#### 测试 CUDA 功能是否正常工作
一旦上述步骤都已完成,就可以拉取一个预构建的支持 CUDA 的 PyTorch 镜像来进行测试。例如:
```bash
docker pull nvidia/cuda:11.7.0-base-ubuntu20.04
```
之后启动一个新的交互式会话并验证 CUDA 是否可用:
```bash
docker run --gpus all -it --rm nvidia/cuda:11.7.0-base-ubuntu20.04 nvidia-smi
```
如果一切顺利的话,应该能看到有关当前 GPU 状态的信息输出,证明 CUDA 已经成功启用了。
阅读全文
相关推荐


















