docker - 如何从 docker-compose.yml 指定 nvidia 运行时？

Question

我可以运行一个 tensorflow 容器并通过以下命令从命令行访问 GPU

$ sudo docker run --runtime=nvidia --rm gcr.io/tensorflow/tensorflow:latest-gpu

我希望能够从 docker-compose 运行这个容器。是否可以从 docker-compose.yml 中指定 --runtime 标志？

Answer 1

目前(2018 年 8 月)，适用于 Docker 的 NVIDIA 容器运行时 (nvidia-docker2) supports Docker Compose .

Yes, use Compose format 2.3 and add runtime: nvidia to your GPU service. Docker Compose must be version 1.19.0 or higher.

示例docker-compose.yml:

version: '2.3'

services:
  nvsmi:
    image: ubuntu:16.04
    runtime: nvidia
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
    command: nvidia-smi

更多 example from NVIDIA blog uses Docker Compose展示如何使用 NVIDIA Container Runtime 启动多个 GPU 容器。