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

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


sudo vim /etc/docker/daemon.json

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

# 重启Docker服务
sudo systemctl restart docker

重启Docker服务

sudo systemctl restart docker

可选:检查重启后的状态(确保服务正常运行)

sudo systemctl status docker

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐