diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml index fe0c9c0d560bd..586109e0fe123 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.