WSL2 NVIDIA Docker 安装教程

在这里插入图片描述

一. 环境

  • WSL ubuntu 22.04。未安装WSL的可参考 WSL 安装
  • nvidia 驱动正常。wsl 终端执行 nvidia-smi
nvidia-smi
Sun Oct  6 20:37:01 2024
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 555.58.02              Driver Version: 556.12         CUDA Version: 12.5     |
|-----------------------------------------+------------------------+----------------------+
| GPU  Name                 Persistence-M | Bus-Id          Disp.A | Volatile Uncorr. ECC |
| Fan  Temp   Perf          Pwr:Usage/Cap |           Memory-Usage | GPU-Util  Compute M. |
|                                         |                        |               MIG M. |
|=========================================+========================+======================|
|   0  NVIDIA GeForce RTX 4080 ...    On  |   00000000:01:00.0 Off |                  N/A |
| N/A   40C    P3             19W /   95W |       0MiB /  12282MiB |      0%      Default |
|                                         |                        |                  N/A 
### 如何在 WSL2安装 NVIDIA Docker 为了在 WSL2 中成功配置并运行带有 GPU 支持的容器环境,需遵循特定步骤来设置必要的组件。由于直接在各 Linux 发行版中单独部署 Docker 不被支持,在此情况下应先通过 Windows 平台安装 Docker Desktop[^1]。 #### 安装 Docker Desktop for Windows 和启用 WSL2 集成 确保已启用了 WSL2 功能,并选择了作为默认版本之后,按照官方指南完成 Docker Desktop 的安装过程。随后访问 **Settings -> Resources -> WSL Integration** 来激活对应选项。 #### 更新 WSL2 内核至最新稳定版本 对于希望利用 GPU 加速特性的开发者而言,保持 WSL2 使用最新的内核至关重要。可以通过微软商店更新到最新发布的长期支持(LTS) 版本或者手动下载适用于 WSL2 的预编译二进制文件进行升级。 #### 设置 CUDA 工具链与驱动程序 针对想要让容器能够识别主机上的 Nvidia 显卡的情况,除了上述准备工作外还需要额外操作: - 确认已经安装了合适的 Nvidia 驱动; - 利用 `choco` 或者其他包管理工具获取所需的 CUDA Toolkit 及 cuDNN 库; - 将这些库路径添加到系统的环境变量当中以便后续调用; #### 安装 nvidia-container-toolkit 接下来就是核心部分——nvidia-container-runtime 的引入。这一步骤允许创建出来的容器继承宿主机所提供的图形计算资源。具体命令如下所示: ```bash # 添加 APT 密钥服务器地址 distribution=$(. /etc/os-release;echo $ID$VERSION_ID) curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list # 执行软件源刷新以及 toolkit 包安装 sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit # 重启 docker 服务使更改生效 sudo systemctl restart docker ``` 此时应该可以顺利启动具备 GPU 能力的支持容器了。测试是否正常工作的一个简单方法是拉取一个预先构建好的镜像并执行其中自带的小型演示应用。 ```bash docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi ``` 如果一切无误,则会看到有关当前系统中可用 GPU 设备的信息列表。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值