windows11安装NVIDIA Container Toolkit便于docker使用NVIDIA 显卡

1、确认 WSL2 已启

打开 PowerShell（以管理员身份运行），输入如下命令：

wsl -l -v

docker-desktop的VERSION 为 2。若是1，如果安装过WSL1和WSL2，可用如下命令切换

wsl --set-default-version 2

若wsl显示不正常，可使用下面命令安装：

wsl --install

2、使用wsl安装ubuntu22.04

安装ubuntu22.04

 wsl --install -d Ubuntu-22.04

如果下载很慢，可尝试在后面加上“–web-download”，如下：

wsl --install -d Ubuntu-22.04 --web-download

安装完成后，会出现输入账号和密码提示。输一个自己能记住的如：yangj、123456。

安装完成后，直接在Windows菜单中搜索Ubuntu，即可以看到Ubuntu 22.04的启动菜单，点击即可进行Ubuntu命令行界面， 可以在WSL的终端界面中操作任何的Linux命令。

Docker不一定要在Windows上安装，安装完Ubuntu后，直接在ubuntu内执行以下命令进行安装：

apt install docker.io

3、验证 Windows GPU 驱动支持 WSL

命令行执行：

nvidia-smi

如果出现正常的版本信息，则表明曾经成功安装过驱动，如提示错误，则需要安装NVIDIA显卡驱动。

驱动正常的情况下，在Windows开始菜单中打开“Ubuntu 22.04″, 进行Linux命令行模式，输入以下命令：

curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
  && curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
    sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
    sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list

因github.io在国内可能会访问不了，所以这一步如果安装失败，则需要借助梯子。

若成功，执行如下命令：

sudo sed -i -e '/experimental/ s/^#//g' /etc/apt/sources.list.d/nvidia-container-toolkit.list

执行apt更新

sudo apt-get update

安装nvidia工具包

sudo apt-get install -y nvidia-container-toolkit

配置 Docker Daemon

sudo nvidia-ctk runtime configure --runtime=docker

若安装了桌面版的docker，点“Setting”->”Docker Engine”，在里面加入如下内容：

"runtimes": {
    "nvidia": {
        "args": [],
        "path": "nvidia-container-runtime"
    }
}

经过以上操作，就可以在Windows命令行下运行带NVidia引擎的容器了。如：

docker run --rm --runtime=nvidia --gpus all ubuntu nvidia-smi