WSL with NVIDIA Container Toolkit

一、wsl 下安装 docker 会提示安装 docekr 桌面版，所以直接安装 docker 桌面版本即可

二、安装 NVIDIA Container Toolkit

NVIDIA Container Toolkit仓库

https://github.com/NVIDIA/nvidia-container-toolkit​github.com/NVIDIA/nvidia-container-toolkit

安装文档

Installing the NVIDIA Container Toolkit​docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html#installation

安卓文档里面的命令，参照标准流程安装即可。也可以参考：Windows11下Docker使用记录（五）_windows docker cuda-CSDN博客

三、docker 追加 NVIDIA runtime 配置

最新配置可以通过命令得到，会输出配置到文件

Installing the NVIDIA Container Toolkit​docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html#configuring-docker

// 最新生成配置命令需要参照链接文档
sudo nvidia-ctk runtime configure --runtime=docker

然后将NVIDIA runtime 配置添加到 docker 的配置中

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

四、简单测试一下

Running a Sample Workload​docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/sample-workload.html#running-a-sample-workload

// 最新测试命令需要参照链接文档
sudo docker run --rm --runtime=nvidia --gpus all ubuntu nvidia-smi