build.yml

name: Build Docker Images

on:
  workflow_dispatch:
    inputs:
      eole_version:
        description: "EOLE version"
        required: true
        type: string
      torch_version:
        description: "PyTorch version"
        required: true
        type: string
      cuda_version:
        description: "CUDA version"
        required: true
        type: string

run-name: ${{ github.workflow }} -- ${{ inputs.eole_version }} torch:${{ inputs.torch_version }}, cuda:${{ inputs.cuda_version }}

env:
  EOLE_VERSION: ${{ inputs.eole_version }}
  TORCH_VERSION: ${{ inputs.torch_version }}
  CUDA_VERSION: ${{ inputs.cuda_version }}

jobs:
  build:
    runs-on: ubuntu-22.04
    # strategy:
    #   matrix:
    #     cuda_version: [11.8.0, 12.1.0]
    permissions: write-all
    steps:
    - name: Maximize build space
      uses: easimon/maximize-build-space@master
      with:
        root-reserve-mb: 512
        swap-size-mb: 10
        remove-dotnet: 'true'
        remove-android: 'true'
        remove-haskell: 'true'
        overprovision-lvm: 'true'
    - name: Checkout repo
      uses: actions/checkout@v4
    - name: Login to ghcr
      uses: docker/login-action@v3
      with:
        registry: ghcr.io
        username: ${{ github.actor }}
        password: ${{ secrets.GITHUB_TOKEN }}
    - name: Build
      run: |
        docker/build.sh ${{ env.EOLE_VERSION }} ${{ env.TORCH_VERSION}} ${{ env.CUDA_VERSION}}