bump: PyTorch to be latest 2.7.1 (#20877)

Author: Borda

.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