SparseML v1.6.0 #14
Workflow file for this run
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Build and Publish Docker Images | |
| on: | |
| release: | |
| types: [published] | |
| jobs: | |
| build-and-push-docker-image: | |
| name: Build and Push Version Tagged Docker Images to GitHub Container Registry | |
| runs-on: ubuntu-latest | |
| permissions: | |
| contents: read | |
| packages: write | |
| steps: | |
| - name: Set up Docker Buildx | |
| if: ${{ startsWith(github.ref, 'refs/tags/v') }} | |
| id: buildx | |
| uses: docker/setup-buildx-action@v2 | |
| with: | |
| buildkitd-flags: --debug | |
| - name: Login to Github Packages | |
| if: ${{ startsWith(github.ref, 'refs/tags/v') }} | |
| uses: docker/login-action@v2 | |
| with: | |
| registry: ghcr.io | |
| username: ${{ github.actor }} | |
| password: ${{ secrets.GITHUB_TOKEN }} | |
| - name: Checkout code | |
| if: ${{ startsWith(github.ref, 'refs/tags/v') }} | |
| uses: actions/checkout@v3 | |
| with: | |
| fetch-depth: 1 | |
| - name: Get Tag | |
| id: extract_tag | |
| run: echo "##[set-output name=tag;]$(echo ${GITHUB_REF_NAME#*/})" | |
| - name: Current Version Name | |
| if: ${{ startsWith(github.ref, 'refs/tags/v') }} | |
| run: | | |
| echo ${{ steps.extract_tag.outputs.tag }} | |
| - name: Build and push sparseml latest using default cuda 11.1.1 | |
| if: ${{ startsWith(github.ref, 'refs/tags/v') }} | |
| uses: docker/build-push-action@v2 | |
| with: | |
| context: ./docker | |
| build-args: | | |
| DEPS=all | |
| VERSION=${{ steps.extract_tag.outputs.tag }} | |
| push: true | |
| tags: | | |
| ghcr.io/neuralmagic/sparseml:latest | |
| - name: Build and push sparseml with all dependencies and default cuda 11.1.1 | |
| if: ${{ startsWith(github.ref, 'refs/tags/v') }} | |
| uses: docker/build-push-action@v2 | |
| with: | |
| context: ./docker | |
| build-args: | | |
| DEPS=all | |
| VERSION=${{ steps.extract_tag.outputs.tag }} | |
| push: true | |
| tags: | | |
| ghcr.io/neuralmagic/sparseml:${{ steps.extract_tag.outputs.tag }} | |
| - name: Build and push sparseml with all dependencies and cuda 10.2 | |
| if: ${{ startsWith(github.ref, 'refs/tags/v') }} | |
| uses: docker/build-push-action@v2 | |
| with: | |
| context: ./docker | |
| build-args: | | |
| DEPS=all | |
| VERSION=${{ steps.extract_tag.outputs.tag }} | |
| CUDA=10.2 | |
| push: true | |
| tags: | | |
| ghcr.io/neuralmagic/sparseml:${{ steps.extract_tag.outputs.tag }}-cu102 | |
| - name: Build and push sparseml-base cu11.1.1 | |
| if: ${{ startsWith(github.ref, 'refs/tags/v') }} | |
| uses: docker/build-push-action@v2 | |
| with: | |
| context: ./docker | |
| build-args: | | |
| DEPS=base | |
| VERSION=${{ steps.extract_tag.outputs.tag }} | |
| push: true | |
| tags: | | |
| ghcr.io/neuralmagic/sparseml-base:${{ steps.extract_tag.outputs.tag }} | |
| - name: Build and push sparseml-base cu10.2 | |
| if: ${{ startsWith(github.ref, 'refs/tags/v') }} | |
| uses: docker/build-push-action@v2 | |
| with: | |
| context: ./docker | |
| build-args: | | |
| DEPS=base | |
| VERSION=${{ steps.extract_tag.outputs.tag }} | |
| CUDA=10.2 | |
| push: true | |
| tags: | | |
| ghcr.io/neuralmagic/sparseml-base:${{ steps.extract_tag.outputs.tag }}-cu10.2 | |
| - name: Image digest | |
| if: ${{ startsWith(github.ref, 'refs/tags/v') }} | |
| run: echo ${{ steps.docker_build.outputs.digest }} |