我们使用Docker的过程中往往需要使用GPU，于是需要安装NVIDIA Driver、NVIDIA Container Toolkit和CUDA Toolkit等工具。

如果你使用的是Ubuntu等Linux原生系统 + docker：你需要在 /etc/docker/daemon.json 中设置default-runtime为nvidia，然后重启docker，确保你启动了NVIDIA Container Toolkit。

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

但是如果你使用的是WSL 2.0 + Docker Desktop，你会发现根本没有 /etc/docker/daemon.json 这个路径。那么这时你需要在Docker Desktop中点击右上角的Setting，然后点击Docker Engine，在这里设置完上述配置后，点击restart，以确保你启动了NVIDIA Container Toolkit。

By the way ~ WSL 2也需要在Resources中设置确保开启。