Bumps stable torch to 2.2.2 (#165)

Lightning-AI · Apr 12, 2024 · f39bea7 · f39bea7
1 parent 0a587b5
commit f39bea7
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/.azure/docker-build.yml b/.azure/docker-build.yml
@@ -41,7 +41,7 @@ jobs:
       matrix:
         # CUDA 12.1
         "cuda 12.1 | torch 2.2 | cudnn FE v1.3":
-          { CUDA_VERSION: "12.1.1", TORCH_VERSION: "2.2.1", TRITON_VERSION: "2.2.0", CUDNN_FRONTEND: "1.3.0" }
+          { CUDA_VERSION: "12.1.1", TORCH_VERSION: "2.2.2", TRITON_VERSION: "2.2.0", CUDNN_FRONTEND: "1.3.0" }
         "cuda 12.1 | torch 2.3 /test | cudnn FE v1.3":
           {
             CUDA_VERSION: "12.1.1",

diff --git a/.azure/gpu-tests.yml b/.azure/gpu-tests.yml
@@ -17,10 +17,10 @@ jobs:
       matrix:
         # CUDA 12.1
         "ubuntu22.04 | cuda 12.1 | python 3.10 | torch 2.2 | regular":
-          docker-image: "ubuntu22.04-cuda12.1.1-cudnn-fe1.3.0-py3.10-pt_2.2.1-apex"
+          docker-image: "ubuntu22.04-cuda12.1.1-cudnn-fe1.3.0-py3.10-pt_2.2.2-apex"
           CUDA_VERSION_MM: "121"
         "ubuntu22.04 | cuda 12.1 | python 3.10 | torch 2.2 | distributed":
-          docker-image: "ubuntu22.04-cuda12.1.1-cudnn-fe1.3.0-py3.10-pt_2.2.1-apex"
+          docker-image: "ubuntu22.04-cuda12.1.1-cudnn-fe1.3.0-py3.10-pt_2.2.2-apex"
           CUDA_VERSION_MM: "121"
           testing: "distributed"
         "ubuntu22.04 | cuda 12.1 | python 3.10 | torch 2.3 | regular":

diff --git a/.azure/notebook-runs.yml b/.azure/notebook-runs.yml
@@ -16,7 +16,7 @@ jobs:
     strategy:
       matrix:
         "ubuntu22.04 | cuda 12.1 | torch 2.2":
-          docker-image: "ubuntu22.04-cuda12.1.1-cudnn-fe1.1.0-py3.10-pt_2.2.1-apex"
+          docker-image: "ubuntu22.04-cuda12.1.1-cudnn-fe1.3.0-py3.10-pt_2.2.2-apex"
           CUDA_VERSION_MM: "121"
         "ubuntu22.04 | cuda 12.1 | torch-nightly":
           docker-image: "ubuntu22.04-cuda12.1.1-cudnn-fe1.3.0-py3.10-pt_main-apex"