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

2023-12-20 20:08:19

在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容器一起使用。

文章来源:https://blog.csdn.net/u011308433/article/details/135115293
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。