From c9b1ffe77c1e22381275f7718526be2f9f06f78d Mon Sep 17 00:00:00 2001 From: Jirka B Date: Mon, 19 May 2025 10:28:51 +0200 Subject: [PATCH 1/2] docker: extend building base docker images for litGPT --- .github/workflows/docker-build.yml | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml index fe0c9c0d560bd..31b4ddbad7602 100644 --- a/.github/workflows/docker-build.yml +++ b/.github/workflows/docker-build.yml @@ -94,6 +94,11 @@ jobs: strategy: fail-fast: false matrix: + # adding dome more images as Thunder mainly using python 3.10, + # and we need to support integrations as for example LitGPT + python-version: ["3.10"] + pytorch-version: ["2.3.1", "2.4.1", "2.5.1", "2.6.0", "2.7.0"] + cuda-version: ["12.4.1"] include: # These are the base images for PL release docker images. # Make sure the matrix here matches the one above. From 41fa9f91b4604bc5a3ea67bb728e6a10e566c2ab Mon Sep 17 00:00:00 2001 From: Jirka B Date: Mon, 19 May 2025 11:42:22 +0200 Subject: [PATCH 2/2] typo lol --- .github/workflows/docker-build.yml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml index 31b4ddbad7602..586109e0fe123 100644 --- a/.github/workflows/docker-build.yml +++ b/.github/workflows/docker-build.yml @@ -96,9 +96,9 @@ jobs: matrix: # adding dome more images as Thunder mainly using python 3.10, # and we need to support integrations as for example LitGPT - python-version: ["3.10"] - pytorch-version: ["2.3.1", "2.4.1", "2.5.1", "2.6.0", "2.7.0"] - cuda-version: ["12.4.1"] + python_version: ["3.10"] + pytorch_version: ["2.3.1", "2.4.1", "2.5.1", "2.6.0", "2.7.0"] + cuda_version: ["12.4.1"] include: # These are the base images for PL release docker images. # Make sure the matrix here matches the one above.