diff --git a/.github/workflows/publish-docker.yml b/.github/workflows/publish-docker.yml index 04a188e..8a63a19 100644 --- a/.github/workflows/publish-docker.yml +++ b/.github/workflows/publish-docker.yml @@ -43,3 +43,15 @@ jobs: cache-to: type=gha,mode=max env: TAG: ${{ startsWith(github.ref, 'refs/tags/') && steps.get-variables.outputs.version || 'latest' }} + + - name: Build and push Cuda Image + uses: docker/build-push-action@v2 + with: + file: ./docker/cuda.Dockerfile + platforms: linux/amd64,linux/arm64 + tags: ${{ steps.get-variables.outputs.gh-username-lower }}/libretranslate:${{ env.TAG }}-cuda + push: true + cache-from: type=gha + cache-to: type=gha,mode=max + env: + TAG: ${{ startsWith(github.ref, 'refs/tags/') && steps.get-variables.outputs.version || 'latest' }}