Skip to content

Build Docker Images -- 0.0.1 torch:2.3.0, cuda:12.1.0 #7

Build Docker Images -- 0.0.1 torch:2.3.0, cuda:12.1.0

Build Docker Images -- 0.0.1 torch:2.3.0, cuda:12.1.0 #7

Workflow file for this run

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}}