ubuntu20.04安装docker-nvidia2

<think>好的，用户想在Ubuntu 20.04上安装支持NVIDIA的Docker，也就是nvidia-docker2。首先，我需要回忆一下安装步骤，并确保所有引用资料中的信息正确无误。 首先，用户需要确保系统已经安装了NVIDIA驱动。根据引用[4]，安装驱动是第一步，所以应该提醒用户检查驱动是否安装正确，可以使用nvidia-smi命令来验证。如果驱动未安装，用户需要先安装合适的驱动，这可能涉及到禁用nouveau驱动或使用ubuntu-drivers工具自动安装。 接下来，安装Docker CE。根据引用[3]，在Ubuntu 20.04上安装Docker-CE社区版需要添加Docker的官方仓库，然后安装特定版本。用户需要更新apt包索引，安装必要的依赖，添加GPG密钥和仓库，然后安装docker-ce。安装完成后，启动并启用docker服务，并将用户加入docker组以避免sudo权限问题。 然后是安装nvidia-docker2。根据引用[2]和引用[4]，需要添加nvidia-docker的仓库，然后安装nvidia-docker2包。安装完成后，重新加载docker守护进程配置，使用sudo pkill -SIGHUP dockerd或者systemctl reload docker。之后，需要配置Docker使用nvidia-container-runtime，这样容器才能访问GPU。修改或创建/etc/docker/daemon.json文件，指定runtime为nvidia。 验证安装是否成功，可以通过运行一个测试容器，比如nvidia-smi或者nvidia/cuda:11.0-base容器，查看GPU信息是否正确输出。 另外，用户可能需要修改docker存储路径或者用户组设置，如引用[4]提到的，但用户当前的问题可能不需要这些步骤，所以可以放在注意事项里简要提及。 需要注意的常见问题包括：驱动版本与CUDA的兼容性、docker守护进程配置是否正确、用户权限问题等。如果用户遇到权限问题，可能需要重新登录或重启系统，让用户组变更生效。 最后，生成相关问题应该围绕安装后的配置、验证、常见问题解决以及应用场景展开，帮助用户进一步了解nvidia-docker的使用和优化。</think>### 在 Ubuntu 20.04 上安装 NVIDIA Docker 的步骤 #### 1. **安装 NVIDIA 驱动** - 确保已安装适用于您的 GPU 的 NVIDIA 驱动。可通过以下命令验证驱动是否安装： ```bash nvidia-smi ``` 若未安装，可使用以下方法安装： ```bash sudo ubuntu-drivers autoinstall # 自动安装推荐驱动 sudo reboot # 重启生效 ``` 或手动选择驱动版本[^4]。 #### 2. **安装 Docker CE** - 添加 Docker 官方仓库并安装 Docker CE： ```bash sudo apt-get update sudo apt-get install apt-transport-https ca-certificates curl software-properties-common curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add - sudo add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu focal stable" sudo apt-get update sudo apt-get install docker-ce # 安装最新版本 ``` - 启动 Docker 服务并加入用户组： ```bash sudo systemctl enable --now docker sudo usermod -aG docker $USER # 将当前用户加入 docker 组 newgrp docker # 刷新用户组（或重新登录） ``` #### 3. **安装 NVIDIA Docker 2** - 添加 NVIDIA Docker 仓库并安装： ```bash distribution=$(. /etc/os-release;echo $ID$VERSION_ID) curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list sudo apt-get update sudo apt-get install -y nvidia-docker2 # 安装 nvidia-docker2 包[^2] ``` - 重启 Docker 服务以加载配置： ```bash sudo systemctl restart docker ``` #### 4. **配置 Docker 使用 NVIDIA Runtime** - 修改 Docker 配置文件 `/etc/docker/daemon.json`，添加以下内容： ```json { "runtimes": { "nvidia": { "path": "/usr/bin/nvidia-container-runtime", "runtimeArgs": [] } }, "default-runtime": "nvidia" } ``` - 重新加载配置： ```bash sudo pkill -SIGHUP dockerd # 或 sudo systemctl reload docker ``` #### 5. **验证安装** - 运行测试容器检查 GPU 是否可用： ```bash docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi ``` 若输出 GPU 信息，则安装成功。 --- ### 注意事项 - **存储路径修改**：如需更改 Docker 镜像存储路径，可修改 `/etc/docker/daemon.json` 中的 `data-root` 字段。 - **兼容性问题**：确保驱动版本与 CUDA 版本匹配，避免容器内应用报错。 ---