150 lines
4.3 KiB
YAML
150 lines
4.3 KiB
YAML
name: CI build Docker
|
|
# https://www.docker.com/blog/first-docker-github-action-is-here
|
|
# https://github.com/docker/build-push-action
|
|
# see: https://help.github.com/en/actions/reference/events-that-trigger-workflows
|
|
on: # Trigger the workflow on push or pull request, but only for the master branch
|
|
push:
|
|
branches: [master, "release/*"] # include release branches like release/1.0.x
|
|
pull_request:
|
|
branches: [master, "release/*"]
|
|
paths:
|
|
- "dockers/**"
|
|
- "!dockers/README.md"
|
|
- "requirements/*.txt"
|
|
- "environment.yml"
|
|
- "requirements.txt"
|
|
- ".github/workflows/*docker*.yml"
|
|
- ".github/workflows/events-nightly.yml"
|
|
- "setup.py"
|
|
|
|
jobs:
|
|
build-PL:
|
|
runs-on: ubuntu-20.04
|
|
strategy:
|
|
fail-fast: false
|
|
matrix:
|
|
python_version: [3.6]
|
|
pytorch_version: [1.4, 1.7]
|
|
steps:
|
|
- name: Checkout
|
|
uses: actions/checkout@v2
|
|
|
|
- name: Build PL Docker
|
|
# publish master/release
|
|
uses: docker/build-push-action@v2
|
|
with:
|
|
build-args: |
|
|
PYTHON_VERSION=${{ matrix.python_version }}
|
|
PYTORCH_VERSION=${{ matrix.pytorch_version }}
|
|
file: dockers/release/Dockerfile
|
|
push: false
|
|
timeout-minutes: 50
|
|
|
|
build-XLA:
|
|
runs-on: ubuntu-20.04
|
|
strategy:
|
|
fail-fast: false
|
|
matrix:
|
|
python_version: [3.7]
|
|
xla_version: [1.6, 1.7, "nightly"]
|
|
steps:
|
|
- name: Checkout
|
|
uses: actions/checkout@v2
|
|
|
|
- name: Build XLA Docker
|
|
# publish master/release
|
|
uses: docker/build-push-action@v2
|
|
with:
|
|
build-args: |
|
|
PYTHON_VERSION=${{ matrix.python_version }}
|
|
XLA_VERSION=${{ matrix.xla_version }}
|
|
file: dockers/base-xla/Dockerfile
|
|
push: false
|
|
timeout-minutes: 50
|
|
|
|
build-cuda:
|
|
runs-on: ubuntu-20.04
|
|
strategy:
|
|
fail-fast: false
|
|
matrix:
|
|
include:
|
|
# todo: see notes in Dockerfile
|
|
- python_version: 3.6
|
|
pytorch_version: 1.4
|
|
- python_version: 3.7
|
|
pytorch_version: 1.6
|
|
- python_version: 3.8
|
|
pytorch_version: 1.7
|
|
# - python_version: 3.9
|
|
# pytorch_version: 1.7
|
|
steps:
|
|
- name: Checkout
|
|
uses: actions/checkout@v2
|
|
|
|
# for PT 1.4 we need to use CUDA 10.1
|
|
- run: |
|
|
cuda=$(python -c "print(10.2 if float(${{matrix.pytorch_version}}) >= 1.5 else 10.1)" 2>&1)
|
|
echo "::set-output name=CUDA::$cuda"
|
|
id: extend
|
|
|
|
- name: Build CUDA Docker
|
|
# publish master/release
|
|
uses: docker/build-push-action@v2
|
|
with:
|
|
build-args: |
|
|
PYTHON_VERSION=${{ matrix.python_version }}
|
|
PYTORCH_VERSION=${{ matrix.pytorch_version }}
|
|
CUDA_VERSION=${{ steps.extend.outputs.CUDA }}
|
|
file: dockers/base-cuda/Dockerfile
|
|
push: false
|
|
timeout-minutes: 50
|
|
|
|
build-conda:
|
|
runs-on: ubuntu-20.04
|
|
strategy:
|
|
fail-fast: false
|
|
matrix:
|
|
include:
|
|
- python_version: 3.6
|
|
pytorch_version: 1.4
|
|
- python_version: 3.7
|
|
pytorch_version: 1.7
|
|
- python_version: 3.8
|
|
pytorch_version: 1.8
|
|
# - python_version: 3.9
|
|
# pytorch_version: 1.8
|
|
steps:
|
|
- name: Checkout
|
|
uses: actions/checkout@v2
|
|
|
|
# for PT 1.3 and 1.4 we need to use CUDA 10.1
|
|
- run: |
|
|
cuda=$(python -c "print(10.2 if float(${{matrix.pytorch_version}}) > 1.4 else 10.1)" 2>&1)
|
|
echo "::set-output name=CUDA::$cuda"
|
|
id: extend
|
|
|
|
- name: Build CUDA Docker
|
|
# publish master/release
|
|
uses: docker/build-push-action@v2
|
|
with:
|
|
build-args: |
|
|
PYTHON_VERSION=${{ matrix.python_version }}
|
|
PYTORCH_VERSION=${{ matrix.pytorch_version }}
|
|
CUDA_VERSION=${{ steps.extend.outputs.CUDA }}
|
|
file: dockers/base-conda/Dockerfile
|
|
push: false
|
|
timeout-minutes: 50
|
|
|
|
build-nvidia:
|
|
runs-on: ubuntu-20.04
|
|
steps:
|
|
- name: Checkout
|
|
uses: actions/checkout@v2
|
|
|
|
- name: Build NVIDIA Docker
|
|
uses: docker/build-push-action@v2
|
|
with:
|
|
file: dockers/nvidia/Dockerfile
|
|
push: false
|
|
timeout-minutes: 50
|