diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml index c7f86a264116c..ef75788679990 100644 --- a/.github/workflows/docker-build.yml +++ b/.github/workflows/docker-build.yml @@ -97,7 +97,7 @@ jobs: # 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.6.0", "2.7.0"] + pytorch_version: ["2.6.0", "2.7.1"] cuda_version: ["12.6.3"] include: # These are the base images for PL release docker images. @@ -108,7 +108,7 @@ jobs: - { python_version: "3.11", pytorch_version: "2.4.1", cuda_version: "12.1.1" } - { python_version: "3.12", pytorch_version: "2.5.1", cuda_version: "12.1.1" } - { python_version: "3.12", pytorch_version: "2.6.0", cuda_version: "12.4.1" } - - { python_version: "3.12", pytorch_version: "2.7.0", cuda_version: "12.6.3" } + - { python_version: "3.12", pytorch_version: "2.7.1", cuda_version: "12.6.3" } steps: - uses: actions/checkout@v4 - uses: docker/setup-buildx-action@v3