builds

Lightning-AI · Mar 27, 2024 · ab83124 · ab83124
1 parent 9d189fc
commit ab83124
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/.azure/gpu-unittests.yml b/.azure/gpu-unittests.yml
@@ -17,11 +17,11 @@ jobs:
   - job: unitest_GPU
     strategy:
       matrix:
-        "PyTorch | 1.10":
+        "PyTorch | 1.10 oldest":
           # Torch does not have build wheels with old Torch versions for newer CUDA
           docker-image: "ubuntu20.04-cuda11.3.1-py3.9-torch1.10"
           torch-ver: "1.10.2"
-        "PyTorch | 1.X":
+        "PyTorch | 1.X LTS":
           docker-image: "ubuntu22.04-cuda11.8.0-py3.9-torch1.13"
           torch-ver: "1.13.1"
         "PyTorch | 2.X stable":

diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml
@@ -67,12 +67,13 @@ jobs:
           # These are the base images for PL release docker images,
           #  so include at least all the combinations in release-dockers.yml.
           - { python: "3.9", pytorch: "1.10", cuda: "11.3.1", ubuntu: "20.04" }
-          - { python: "3.9", pytorch: "1.11", cuda: "11.8.0", ubuntu: "22.04" }
+          #- { python: "3.9", pytorch: "1.11", cuda: "11.8.0", ubuntu: "22.04" }
           - { python: "3.9", pytorch: "1.13", cuda: "11.8.0", ubuntu: "22.04" }
-          - { python: "3.10", pytorch: "2.0", cuda: "11.8.0", ubuntu: "22.04" }
-          - { python: "3.10", pytorch: "2.0", cuda: "12.1.1", ubuntu: "22.04" }
-          - { python: "3.10", pytorch: "2.1", cuda: "12.1.1", ubuntu: "22.04" }
-          - { python: "3.11", pytorch: "2.1", cuda: "12.1.1", ubuntu: "22.04" }
+          #- { python: "3.10", pytorch: "2.0", cuda: "11.8.0", ubuntu: "22.04" }
+          #- { python: "3.10", pytorch: "2.0", cuda: "12.1.1", ubuntu: "22.04" }
+          #- { python: "3.10", pytorch: "2.1", cuda: "12.1.1", ubuntu: "22.04" }
+          #- { python: "3.11", pytorch: "2.1", cuda: "12.1.1", ubuntu: "22.04" }
+          - { python: "3.10", pytorch: "2.2", cuda: "12.1.1", ubuntu: "22.04" }
           - { python: "3.11", pytorch: "2.2", cuda: "12.1.1", ubuntu: "22.04" }
           - { python: "3.11", pytorch: "2.3", cuda: "12.1.1", ubuntu: "22.04" }
     steps: