Docker Compose 共享 GPU

Docker Compose 是 Docker 提供的一个用于定义和运行多个容器的工具。它允许开发人员通过一个单独的配置文件来定义多个相关联的容器，简化了容器集群的管理和部署。

在一些需要使用 GPU 运行的应用场景中，你可能想要在多个容器之间共享一个或多个 GPU 设备。这种共享 GPU 资源的需求在深度学习、图形渲染等场景中尤为常见。通过 Docker Compose，你可以轻松地配置和管理共享 GPU 资源的容器环境。

步骤1：安装 Docker 和 Docker Compose

首先，你需要在你的开发机上安装 Docker 和 Docker Compose。可以根据你的操作系统类型，查找并按照对应的安装指南进行安装。

• Docker 安装指南：https://docs.docker.com/get-docker/
• Docker Compose 安装指南：https://docs.docker.com/compose/install/

步骤2：配置 Docker Compose 文件

创建一个名为 docker-compose.yml 的文件，并按照以下格式进行配置：

```yaml
version: '3'
services:
  your_service_name:
    build: .
    runtime: nvidia
    environment:
      - NVIDIA_VISIBLE_DEVICES=all

请将 `your_service_name` 替换为你的服务名称，将 `build` 替换为你自己的构建配置。这里的 `runtime: nvidia` 指定了运行时为 `nvidia`，以便 Docker Compose 在运行容器时能够访问到 GPU 资源。`NVIDIA_VISIBLE_DEVICES=all` 表示将所有可用的 GPU 设备分配给该容器。

#### 步骤3：构建和运行容器

运行以下命令来构建和运行容器：

```bash
docker-compose up -d

这将使用你的配置文件来构建和启动容器。

步骤4：验证 GPU 共享

在容器内部，可以通过以下命令来验证 GPU 是否已经成功共享：

nvidia-smi

如果显示了 GPU 的相关信息，则说明 GPU 资源已经成功共享给容器。

总结

通过 Docker Compose，你可以方便地配置和管理共享 GPU 资源的容器环境。这为开发人员在需要 GPU 加速的应用场景中提供了更高的灵活性和方便性。

参考链接：

• https://docs.docker.com/compose/
• https://github.com/NVIDIA/nvidia-docker