Add docker release workflow (#1743)

Author: oelachqar

.github/workflows/release_docker.yaml

@@ -0,0 +1,84 @@
+name: Release - Docker
+
+on:
+  release:
+    types: [published]
+  workflow_dispatch:
+    inputs:
+      tag:
+        description: 'Docker image tag (defaults to release tag or "manual")'
+        required: false
+        type: string
+  pull_request:
+    paths:
+      - 'Dockerfile'
+
+env:
+  REGISTRY: ghcr.io
+  IMAGE_NAME: ${{ github.repository }}
+
+jobs:
+  build-and-push:
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+      packages: write
+      id-token: write
+
+    steps:
+    - name: Checkout repository
+      uses: actions/checkout@v4
+      with:
+        # For releases, checkout the release tag; for workflow_dispatch, checkout the latest commit
+        ref: ${{ github.event_name == 'release' && github.event.release.tag_name || github.sha }}
+
+    - name: Display checkout info
+      run: |
+        echo "Checked out commit: $(git rev-parse HEAD)"
+        echo "Event name: ${{ github.event_name }}"
+        echo "Release tag: ${{ github.event.release.tag_name }}"
+        echo "GitHub SHA: ${{ github.sha }}"
+
+    - # Add support for more platforms with QEMU (optional)
+      # https://github.com/docker/setup-qemu-action
+      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 GitHub Container Registry
+      uses: docker/login-action@v3
+      with:
+        registry: ${{ env.REGISTRY }}
+        username: ${{ github.actor }}
+        password: ${{ secrets.GITHUB_TOKEN }}
+
+    - name: Extract metadata
+      id: meta
+      uses: docker/metadata-action@v5
+      with:
+        images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
+        tags: |
+          type=ref,event=branch
+          type=ref,event=pr
+          type=semver,pattern={{version}}
+          type=semver,pattern={{major}}.{{minor}}
+          type=semver,pattern={{major}}
+          type=raw,value=${{ inputs.tag }},enable=${{ inputs.tag != '' }}
+          type=raw,value=latest,enable=${{ github.event_name == 'release' }}
+
+    - name: Build and push Docker image
+      uses: docker/build-push-action@v5
+      with:
+        context: .
+        file: Dockerfile
+        platforms: linux/amd64
+        push: true
+        tags: ${{ steps.meta.outputs.tags }}
+        labels: ${{ steps.meta.outputs.labels }}
+        build-args: |
+          TARGETPLATFORM=linux/amd64
+          OUMI_VERSION=${{ github.event.release.tag_name }}
+        cache-from: type=gha
+        cache-to: type=gha,mode=max

Dockerfile

@@ -1,6 +1,8 @@
 ARG TARGETPLATFORM=linux/amd64
 FROM --platform=${TARGETPLATFORM} pytorch/pytorch:2.5.1-cuda12.4-cudnn9-runtime
 
+# ARG for oumi version - defaults to empty string which will install latest
+ARG OUMI_VERSION=
 
 WORKDIR /oumi_workdir
 
@@ -23,8 +25,13 @@ RUN apt-get update && \
 
 
 # Install Oumi dependencies
+# If OUMI_VERSION is provided, install that specific version, otherwise install latest
 RUN pip install --no-cache-dir uv && \
-    uv pip install --system --no-cache-dir "oumi[gpu]"
+    if [ -z "$OUMI_VERSION" ]; then \
+        uv pip install --system --no-cache-dir "oumi[gpu]"; \
+    else \
+        uv pip install --system --no-cache-dir "oumi[gpu]==$OUMI_VERSION"; \
+    fi
 
 # Switch to oumi user
 USER oumi