Create GH action for automated docker builds on releases (#1559)

* Create docker_builds.yml * Update docker_builds.yml * Update docker_builds.yml * Update docker_builds.yml * Update docker_builds.yml
Lightning-AI · Apr 26, 2020 · 13bf772 · 13bf772
1 parent 17bce62
commit 13bf772
Showing 1 changed file with 46 additions and 0 deletions.
diff --git a/.github/workflows/docker_builds.yml b/.github/workflows/docker_builds.yml
@@ -0,0 +1,46 @@
+name: Publish Docker
+on:
+  push: 
+    branches:
+      - master 
+  release: 
+    types: 
+       - created
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python_version: [3.6, 3.7, 3.8]
+        pytorch_version: [1.1, 1.2, 1.3, 1.4, 1.5]
+    steps:
+    - name: Extract Current Tag
+      if: contains(github.ref, 'refs/tags/')
+      id: get_version
+      run: echo ::set-output name=VERSION::${GITHUB_REF/refs\/tags\//}
+    - uses: actions/checkout@2.14
+    - name: Publish Releases to Docker
+      # only on releases
+      uses: elgohr/Publish-Docker-Github-Action@master
+      if: contains(github.ref, 'refs/tags/') && !contains(${{ steps.get_version.outputs.VERSION }}, 'rc') %% !contains(${{ steps.get_version.outputs.VERSION }}, 'dev')
+      with:
+        name: pytorchlightning/pytorch_lightning
+        username: ${{ secrets.DOCKER_USERNAME }}
+        password: ${{ secrets.DOCKER_PASSWORD }}
+        dockerfile: docker/Dockerfile
+        buildargs: PYTHON_VERSION=${{ matrix.python_version }},PYTORCH_VERSION=${{ matrix.pytorch_version }},LIGHTNING_VERSION=${{ steps.get_version.outputs.VERSION }}
+        tags: "${{ steps.get_version.outputs.VERSION }}_py${{ matrix.python_version }}_torch${{ matrix.pytorch_version }},stable_py${{ matrix.python_version }}_torch${{ matrix.pytorch_version }}"
+    - name: Publish Master
+      # publish master
+      uses: elgohr/Publish-Docker-Github-Action@master
+      if: github.event_name == 'push'
+      with:
+        name: pytorchlightning/pytorch_lightning
+        username: ${{ secrets.DOCKER_USERNAME }}
+        password: ${{ secrets.DOCKER_PASSWORD }}
+        dockerfile: docker/Dockerfile
+        buildargs: PYTHON_VERSION=${{ matrix.python_version }},PYTORCH_VERSION=${{ matrix.pytorch_version }},LIGHTNING_VERSION=${{ steps.get_version.outputs.VERSION }}
+        tags: "latest_py${{ matrix.python_version }}_torch${{ matrix.pytorch_version }}"
+
+