diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml index ec1bf23..aea7117 100644 --- a/.github/workflows/CI.yml +++ b/.github/workflows/CI.yml @@ -130,3 +130,36 @@ jobs: with: command: upload args: --non-interactive --skip-existing wheels-*/* + + push-docker: + runs-on: ubuntu-latest + permissions: + contents: read + packages: write + strategy: + matrix: + tag: [cpu, cuda] + platform: + - linux/amd64 + - linux/arm64 + steps: + - 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: Login to GitHub Container Registry + uses: docker/login-action@v3 + with: + registry: ghcr.io + username: ${{ github.actor }} + password: ${{ secrets.GITHUB_TOKEN }} + - name: Build and push image + uses: docker/build-push-action@v6 + with: + context: . + push: true + tags: | + ghcr.io/${{ github.repository }}:${{ matrix.tag }} + file: docker/${{ matrix.tag }}.Dockerfile + platforms: ${{ matrix.platform }}