publish.yml

name: Publish

on:
  push:
    branches:
      - main
      - master

jobs:
  publish:
    runs-on: ubuntu-latest

    strategy:
      matrix:
        python-version: ["3.8", "3.9", "3.10", "3.11", "3.12"]
        cuda-version: ["11.8"]

    name: Python ${{ matrix.python-version }} CUDA ${{ matrix.cuda-version }}

    steps:
    - name: Checkout
      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 Docker Hub
      uses: docker/login-action@v3 
      with:
        username: ${{ secrets.REGISTRY_USERNAME }}
        password: ${{ secrets.REGISTRY_PASSWORD }}

    - name: Build and push Docker image
      uses: docker/build-push-action@v5
      with:
        build-args: |
          "PYTHON_VERSION=${{ matrix.python-version }}"
          "CUDA_VERSION=${{ matrix.cuda-version }}"
        context: .
        platforms: linux/amd64,linux/arm64
        push: true
        tags: superlinear/python-gpu:${{ matrix.python-version }}-cuda${{ matrix.cuda-version }}