【Ubuntu】为Docker安装NVIDIA运行时环境（NVIDIA Container Runtime for Docker）

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

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

   首先，确保您的系统已经安装了Docker。您可以使用以下命令进行安装：

   sudo apt update
   sudo apt install docker.io
   

2. 添加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
   

3. 安装NVIDIA容器运行时：

   更新Apt包列表并安装NVIDIA容器运行时：

   sudo apt update
   sudo apt install nvidia-container-runtime
   

4. 配置Docker以使用NVIDIA运行时：

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

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

   保存并关闭文件。

5. 重新启动Docker服务：

   sudo systemctl restart docker
   

6. 验证NVIDIA运行时的安装：

   使用以下命令运行一个测试容器来验证NVIDIA运行时的安装：

   docker run --gpus all nvidia/cuda:11.0-base nvidia-smi
   

   如果一切正常，您应该能够看到NVIDIA的nvidia-smi命令的输出，显示了GPU的详细信息。

这样，您的Ubuntu系统上的Docker就已配置为使用NVIDIA容器运行时了。确保您的NVIDIA显卡驱动和CUDA工具包也已正确安装，以便与Docker容器一起使用。