[Feat] Add arm docker image #114
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
| # https://docs.github.com/en/actions/publishing-packages/publishing-docker-images#publishing-images-to-github-packages | |
| name: Docker image on ghcr.io | |
| on: | |
| push: | |
| tags: | |
| - 'v*' | |
| pull_request: | |
| branches: main | |
| schedule: | |
| - cron: '0 2 1 6 *' # At 02:00 on day-of-month 1 in June (once a year actually) | |
| env: | |
| REGISTRY: ghcr.io | |
| jobs: | |
| build-and-push-image: | |
| runs-on: ubuntu-latest | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| image: ["ubuntu:24.04", "nvidia/cuda:12.6.2-base-ubuntu24.04"] | |
| # Must match version at https://www.python.org/ftp/python/ | |
| python: [3.10.13] | |
| platform: | |
| - linux/amd64 | |
| - linux/arm64 # >= Raspberry Pi 4 | |
| # Sets the permissions granted to the `GITHUB_TOKEN` for the actions in this job. | |
| permissions: | |
| contents: read | |
| packages: write | |
| steps: | |
| - name: Checkout repository | |
| uses: actions/checkout@v4 | |
| - name: Set up QEMU | |
| uses: docker/setup-qemu-action@v3 | |
| - name: Set up Docker Buildx | |
| uses: docker/setup-buildx-action@v3 | |
| - name: Log in to the Container registry | |
| uses: docker/login-action@v3 | |
| with: | |
| registry: ${{ env.REGISTRY }} | |
| username: ${{ github.actor }} | |
| password: ${{ secrets.GITHUB_TOKEN }} | |
| - name: Sanitize docker tag | |
| run: | | |
| ARCH="${{ matrix.platform == 'linux/amd64' && 'amd64' || (matrix.platform == 'linux/arm64' && 'arm64') }}" | |
| PREFIX_DOCKER_TAG="OnnxTR-${ARCH}-${{ matrix.image == 'ubuntu:24.04' && 'cpu' || 'gpu' }}-py${{ matrix.python }}-" | |
| PREFIX_DOCKER_TAG=$(echo $PREFIX_DOCKER_TAG | sed 's/,/-/g') | |
| echo PREFIX_DOCKER_TAG=${PREFIX_DOCKER_TAG} >> $GITHUB_ENV | |
| - name: Extract metadata (tags, labels) for Docker | |
| id: meta | |
| uses: docker/metadata-action@v5 | |
| with: | |
| images: ${{ env.REGISTRY }}/${{ github.repository }} | |
| tags: | | |
| # used only on schedule event | |
| type=schedule,pattern={{date 'YYYY-MM'}},prefix=${{ env.PREFIX_DOCKER_TAG }} | |
| # used only if a tag following semver is published | |
| type=semver,pattern={{raw}},prefix=${{ env.PREFIX_DOCKER_TAG }} | |
| - name: Build Docker image | |
| id: build | |
| uses: docker/build-push-action@v6 | |
| with: | |
| context: . | |
| platforms: ${{ matrix.platform }} | |
| # Set SYSTEM to 'cpu-headless' if the image is ubuntu:24.04 | |
| # and 'gpu-headless' otherwise | |
| build-args: | | |
| BASE_IMAGE=${{ matrix.image }} | |
| SYSTEM=${{ matrix.image == 'ubuntu:24.04' && 'cpu-headless' || 'gpu-headless' }} | |
| PYTHON_VERSION=${{ matrix.python }} | |
| ONNXTR_REPO=${{ github.repository }} | |
| ONNXTR_VERSION=${{ github.sha }} | |
| push: false # Push only if `import onnxtr` works | |
| tags: ${{ steps.meta.outputs.tags }} | |
| - name: Check if `import onnxtr` works | |
| run: docker run --platform=${{ matrix.platform }} ${{ steps.build.outputs.imageid }} python3 -c 'import onnxtr; print(onnxtr.__version__)' | |
| - name: Inspect the image | |
| run: docker buildx imagetools inspect ${{ steps.build.outputs.imageid }} | |
| - name: Push Docker image | |
| # Push only if the CI is not triggered by "PR on main" | |
| if: ${{ (github.ref == 'refs/heads/main' && github.event_name != 'pull_request') || (startsWith(github.ref, 'refs/tags') && github.event_name == 'push') }} | |
| uses: docker/build-push-action@v6 | |
| with: | |
| context: . | |
| platforms: ${{ matrix.platform }} | |
| # Set SYSTEM to 'cpu-headless' if the image is ubuntu:24.04 | |
| # and 'gpu-headless' otherwise | |
| build-args: | | |
| BASE_IMAGE=${{ matrix.image }} | |
| SYSTEM=${{ matrix.image == 'ubuntu:24.04' && 'cpu-headless' || 'gpu-headless' }} | |
| PYTHON_VERSION=${{ matrix.python }} | |
| ONNXTR_REPO=${{ github.repository }} | |
| ONNXTR_VERSION=${{ github.sha }} | |
| push: true | |
| tags: ${{ steps.meta.outputs.tags }} |