diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml
index 586109e0fe123..c7f86a264116c 100644
--- a/.github/workflows/docker-build.yml
+++ b/.github/workflows/docker-build.yml
@@ -97,8 +97,8 @@ 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.3.1", "2.4.1", "2.5.1", "2.6.0", "2.7.0"]
-        cuda_version: ["12.4.1"]
+        pytorch_version: ["2.6.0", "2.7.0"]
+        cuda_version: ["12.6.3"]
         include:
           # These are the base images for PL release docker images.
           # Make sure the matrix here matches the one above.