检查WSL2-ubuntu本身, 是否使用了本机的显卡

1
nvidia-smi

更新软件包列表

1
sudo apt update && sudo apt upgrade -y

添加 NVIDIA Container Toolkit GPG 密钥

1
curl -s -L https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg

可能需要使用代理才能正确添加

1
curl -x socks5h://localhost:7448 -s -L https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg

添加 NVIDIA Container Toolkit 仓库

1
2
3
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

可能需要使用代理才能正确添加

1
2
3
curl -x socks5h://localhost:7448 -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

安装 NVIDIA Container Toolkit

  • 更新软件包列表：

    1
    sudo apt-get update

    or

    1
    sudo apt update && sudo apt upgrade -y

  • 安装 NVIDIA Container Toolkit：

    如果显示找不到, 则是因为apt update的更新失败, 需重新执行更新命令直到成功后再进行。 

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

  • 配置运行时：

    1
    sudo nvidia-ctk runtime configure --runtime=docker

重启 Docker 并验证安装

  • 重启 Docker：

    1
    sudo systemctl restart docker

  • 在需要使用显卡直通的测试容器内验证安装：

    1
    nvidia-smi

这是在 Ubuntu 系统上安装和设置 NVIDIA Container Toolkit 的一般过程。

由于国内网络问题, 在执行apt-get update时, 可能会存在失败的问题。因此需要耐心的多尝试几次, 碰上网络好的时候很快就能成功了。

参考

https://docs.qq.com/aio/DSXd3a1RmaFRTZXBP?p=GmMdmSP5UNhl22Xmp3Tgyx&client_hint=0