docker.stream_manager.gpu.yml

name: Build and Push Container with Stream Manager GPU

on:
  release:
    types: [created]
  push:
    branches: [ main ]
  workflow_dispatch:
    inputs:
      force_push:
        type: boolean
        description: "Do you want to push image after build?"
        default: false

env:
  VERSION: '0.0.0' # Default version, will be overwritten

jobs:
  docker:
    runs-on: ubuntu-latest
    steps:
      - name: Remove unnecessary files
        run: |
          sudo rm -rf /usr/share/dotnet
          sudo rm -rf "$AGENT_TOOLSDIRECTORY"
      -
        name: Set up QEMU
        uses: docker/setup-qemu-action@v2
      -
        name: Set up Docker Buildx
        uses: docker/setup-buildx-action@v2
      -
        name: Login to Docker Hub
        uses: docker/login-action@v2
        with:
          username: ${{ secrets.DOCKERHUB_USERNAME }}
          password: ${{ secrets.DOCKERHUB_TOKEN }}
      - 
        name: 🛎️ Checkout
        uses: actions/checkout@v3
      -
        name: Read version from file
        run: echo "VERSION=$(DISABLE_VERSION_CHECK=true python ./inference/core/version.py)" >> $GITHUB_ENV
      -
        name: Build and Push
        uses: docker/build-push-action@v4
        with:
          push: ${{ github.event_name == 'release' || (github.event.inputs.force_push == 'true')}}
          tags: roboflow/roboflow-inference-stream-manager-gpu:latest,roboflow/roboflow-inference-stream-manager-gpu:${{env.VERSION}}
          cache-from: type=registry,ref=roboflow/roboflow-inference-stream-manager-gpu:cache
          cache-to: type=registry,ref=roboflow/roboflow-inference-stream-manager-gpu,mode=max
          platforms: linux/amd64
          file: ./docker/dockerfiles/Dockerfile.onnx.gpu.stream_manager