ci/docker/docker-compose.yml

# -*- mode: dockerfile -*-
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.
#
# We use the cache_from feature introduced in file form version 3.4 (released 2017-11-01)
version: "3.4"

# For simplicity, only the centos7_cpu is commented. But the comments apply to
# all other services as well.
services:
  ###################################################################################################
  # Dockerfile.build.centos7 based images used for building on CentOS7. On
  # CentOS7, we respectively test the oldest supported toolchain and dependency
  # versions
  ###################################################################################################
  centos7_cpu:
    # The resulting image will be named build.centos7_cpu and will be
    # pushed to the dockerhub user specified in the environment variable
    # ${DOCKER_CACHE_REGISTRY} (typicall "mxnetci") under this name
    image: ${DOCKER_CACHE_REGISTRY}:build.centos7_cpu
    build:
      context: .
      dockerfile: Dockerfile.build.centos7
      # Use "base" target declared in Dockerfile.build.centos7 as "build.centos7_cpu:latest"  
      target: base
      args:
        # BASE_IMAGE is used to dynamically specify the FROM image in Dockerfile.build.centos7
        BASE_IMAGE: centos:7
      cache_from:
        - ${DOCKER_CACHE_REGISTRY}:build.centos7_cpu
  centos7_gpu_cu100:
    image: ${DOCKER_CACHE_REGISTRY}:build.centos7_gpu_cu100
    build:
      context: .
      dockerfile: Dockerfile.build.centos7
      target: base
      args:
        BASE_IMAGE: nvidia/cuda:10.0-cudnn7-devel-centos7
      cache_from:
        - ${DOCKER_CACHE_REGISTRY}:build.centos7_gpu_cu100
  centos7_gpu_cu101:
    image: ${DOCKER_CACHE_REGISTRY}:build.centos7_gpu_cu101
    build:
      context: .
      dockerfile: Dockerfile.build.centos7
      target: base
      args:
        BASE_IMAGE: nvidia/cuda:10.1-cudnn7-devel-centos7
      cache_from:
        - ${DOCKER_CACHE_REGISTRY}:build.centos7_gpu_cu101
  centos7_gpu_cu102:
    image: ${DOCKER_CACHE_REGISTRY}:build.centos7_gpu_cu102
    build:
      context: .
      dockerfile: Dockerfile.build.centos7
      target: base
      args:
        BASE_IMAGE: nvidia/cuda:10.2-cudnn8-devel-centos7
      cache_from:
        - ${DOCKER_CACHE_REGISTRY}:build.centos7_gpu_cu102
  centos7_gpu_cu110:
    image: ${DOCKER_CACHE_REGISTRY}:build.centos7_gpu_cu110
    build:
      context: .
      dockerfile: Dockerfile.build.centos7
      target: base
      args:
        BASE_IMAGE: nvidia/cuda:11.0-cudnn8-devel-centos7
      cache_from:
        - ${DOCKER_CACHE_REGISTRY}:build.centos7_gpu_cu110
  centos7_gpu_cu112:
    image: ${DOCKER_CACHE_REGISTRY}:build.centos7_gpu_cu112
    build:
      context: .
      dockerfile: Dockerfile.build.centos7
      target: base
      args:
        BASE_IMAGE: nvidia/cuda:11.2.1-cudnn8-devel-centos7
      cache_from:
        - ${DOCKER_CACHE_REGISTRY}:build.centos7_gpu_cu112