【Ubuntu】为Docker安装NVIDIA运行时环境（NVIDIA Container Runtime for Docker）-Toy模板网

这篇具有很好参考价值的文章主要介绍了【Ubuntu】为Docker安装NVIDIA运行时环境（NVIDIA Container Runtime for Docker）。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

在Ubuntu系统上，为Docker安装NVIDIA运行时环境（NVIDIA Container Runtime for Docker）通常涉及以下步骤：

安装Docker（如果您还没有安装）：

首先，确保您的系统已经安装了Docker。您可以使用以下命令进行安装：
```
sudo apt update
sudo apt install docker.io
```

添加NVIDIA容器运行时的存储库：

执行以下命令以添加NVIDIA的Apt存储库：

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

安装NVIDIA容器运行时：

更新Apt包列表并安装NVIDIA容器运行时：
```
sudo apt update
sudo apt install nvidia-container-runtime
```

配置Docker以使用NVIDIA运行时：

编辑或创建Docker的配置文件（如果不存在）/etc/docker/daemon.json，并添加以下内容：

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

保存并关闭文件。

重新启动Docker服务：
```
sudo systemctl restart docker
```
验证NVIDIA运行时的安装：

使用以下命令运行一个测试容器来验证NVIDIA运行时的安装：
```
docker run --gpus all nvidia/cuda:11.0-base nvidia-smi
```
如果一切正常，您应该能够看到NVIDIA的nvidia-smi命令的输出，显示了GPU的详细信息。