### 在 Docker Compose YAML 文件中指定使用 P40 或 H20 显卡的方法 在 Docker Compose 中，可以通过 `deploy` 部分的 `resources` 字段或直接使用 `--gpus` 参数来指定容器使用的显卡。以下是如何在 `docker-compose.yml` 文件中配置容器以使用特定 GPU（如 P40 或 H20）的详细方法。 #### 1. 使用 `deploy` 配置 GPU 资源 Docker Compose 支持通过 `deploy` 部分定义资源限制，包括 GPU 分配。以下是示例配置： ```yaml version: '3.8' services: gpu_service: image: nvidia/cuda:11.7-base deploy: resources: reservations: devices: - capabilities: [gpu] driver: nvidia count: 1 specifications: vendor: "nvidia.com" model: "Tesla P40" # 指定显卡型号为 P40 command: ["nvidia-smi"] ``` 上述配置通过 `devices` 字段指定了 GPU 的数量和型号[^3]。`specifications` 部分可以进一步细化到具体的 GPU 型号或 UUID。 #### 2. 使用 `runtime` 和 `device` 指定 GPU 另一种方式是通过 `runtime` 和 `device` 字段直接指定 GPU 设备。例如： ```yaml version: '3.8' services: gpu_service: image: nvidia/cuda:11.7-base runtime: nvidia deploy: resources: reservations: devices: - driver: nvidia device_ids: ["0"] # 指定 GPU 索引号为 0（假设 P40 是索引 0） command: ["nvidia-smi"] ``` 在此配置中，`device_ids` 字段用于明确指定 GPU 的索引号。可以通过运行 `nvidia-smi` 命令查看主机上的 GPU 列表及其索引号[^3]。 #### 3. 使用 `--gpus` 参数 如果需要更灵活地控制 GPU 分配，可以在启动容器时通过命令行参数指定 GPU。例如，在 `docker-compose.yml` 文件中添加 `command` 或 `entrypoint` 来传递 `--gpus` 参数： ```yaml version: '3.8' services: gpu_service: image: nvidia/cuda:11.7-base command: ["sh", "-c", "nvidia-container-runtime --gpus 'device=0' && nvidia-smi"] ``` 此方法适用于需要动态分配 GPU 的场景。注意，`nvidia-container-runtime` 必须已正确安装并配置[^5]。 #### 4. 验证 GPU 分配 为了确保容器正确使用了指定的 GPU，可以在容器内运行 `nvidia-smi` 命令进行验证。例如： ```bash docker exec <container_id> nvidia-smi ``` 输出应显示分配给容器的 GPU 信息，包括型号和使用情况[^3]。 --- ###