docker-compose.test.yml

#
# DO NOT MANUALLY EDIT THIS FILE
# AUTOGENERATED BY build/ci_matrix.py
#

version: '2.3'
services:
  test-base:
    build:
      context: .
      dockerfile: build/neuropod.dockerfile
      target: neuropod-base
    privileged: true

  test-gpu:
    extends: test-base
    build:
      args:
        NEUROPOD_IS_GPU: "true"
    runtime: nvidia

  test-cpu-variant-tf-1_12_0-torch-1_1_0-py2_7:
    extends: test-base
    build:
      args:
        NEUROPOD_TENSORFLOW_VERSION: 1.12.0
        NEUROPOD_TORCH_VERSION: 1.1.0
        NEUROPOD_PYTHON_VERSION: 2.7

  test-cpu-variant-tf-1_13_1-torch-1_2_0-py3_5:
    extends: test-base
    build:
      args:
        NEUROPOD_TENSORFLOW_VERSION: 1.13.1
        NEUROPOD_TORCH_VERSION: 1.2.0
        NEUROPOD_PYTHON_VERSION: 3.5

  test-cpu-variant-tf-1_14_0-torch-1_3_0-py3_6:
    extends: test-base
    build:
      args:
        NEUROPOD_TENSORFLOW_VERSION: 1.14.0
        NEUROPOD_TORCH_VERSION: 1.3.0
        NEUROPOD_PYTHON_VERSION: 3.6

  test-cpu-variant-tf-1_15_0-torch-1_4_0-py3_7:
    extends: test-base
    build:
      args:
        NEUROPOD_TENSORFLOW_VERSION: 1.15.0
        NEUROPOD_TORCH_VERSION: 1.4.0
        NEUROPOD_PYTHON_VERSION: 3.7

  test-cpu-variant-tf-2_2_0-torch-1_5_0-py3_8:
    extends: test-base
    build:
      args:
        NEUROPOD_TENSORFLOW_VERSION: 2.2.0
        NEUROPOD_TORCH_VERSION: 1.5.0
        NEUROPOD_PYTHON_VERSION: 3.8

  test-gpu-variant-tf-1_12_0-torch-1_1_0-py2_7:
    extends: test-gpu
    build:
      args:
        NEUROPOD_CUDA_VERSION: 9.0
        NEUROPOD_TENSORFLOW_VERSION: 1.12.0
        NEUROPOD_TORCH_VERSION: 1.1.0
        NEUROPOD_PYTHON_VERSION: 2.7

  test-gpu-variant-tf-1_13_1-torch-1_2_0-py3_5:
    extends: test-gpu
    build:
      args:
        NEUROPOD_CUDA_VERSION: 10.0
        NEUROPOD_TENSORFLOW_VERSION: 1.13.1
        NEUROPOD_TORCH_VERSION: 1.2.0
        NEUROPOD_PYTHON_VERSION: 3.5

  test-gpu-variant-tf-1_14_0-torch-1_3_0-py3_6:
    extends: test-gpu
    build:
      args:
        NEUROPOD_CUDA_VERSION: 10.0
        NEUROPOD_TENSORFLOW_VERSION: 1.14.0
        NEUROPOD_TORCH_VERSION: 1.3.0
        NEUROPOD_PYTHON_VERSION: 3.6

  test-gpu-variant-tf-1_15_0-torch-1_4_0-py3_7:
    extends: test-gpu
    build:
      args:
        NEUROPOD_CUDA_VERSION: 10.0
        NEUROPOD_TENSORFLOW_VERSION: 1.15.0
        NEUROPOD_TORCH_VERSION: 1.4.0
        NEUROPOD_PYTHON_VERSION: 3.7

  test-gpu-variant-tf-2_2_0-torch-1_5_0-py3_8:
    extends: test-gpu
    build:
      args:
        NEUROPOD_CUDA_VERSION: 10.1
        NEUROPOD_TENSORFLOW_VERSION: 2.2.0
        NEUROPOD_TORCH_VERSION: 1.5.0
        NEUROPOD_PYTHON_VERSION: 3.8