cuddly-mechanic-74177
08/15/2024, 2:01 PMstrategy:
matrix:
python_version:
- '3.11'
- '3.12'
cuda_version:
- '12.4.1-cudnn'
- '12.5.1-cudnn'
image_type:
- runtime
- devel
uses: XXXXXXX/.github/workflows/build-docker-image.yaml@v8
with:
dockerfile: base/nvidia/Dockerfile
images: XXXXXXX/${{ github.event.repository.name }}/cuda-${{matrix.image_type}}-${{ matrix.cuda_version }}-${{ matrix.python_version }}
context: base/nvidia
build-args: |
PYTHON_VERSION=${{ matrix.python_version }}
CUDA_VERSION=${{ matrix.cuda_version }}
TYPE=${{matrix.image_type}}
cuddly-mechanic-74177
08/15/2024, 2:02 PMdocker_image(
extra_build_args=parametrize(runtime=["TYPE=runtime"], devel=["TYPE=devel"])
)
cuddly-mechanic-74177
08/15/2024, 2:02 PMcuddly-mechanic-74177
08/15/2024, 2:03 PM