bump to pytorch 2.3.1 (#847)

Lightning-AI · Jul 25, 2024 · 1ff7e2d · 1ff7e2d
1 parent c6bdb09
commit 1ff7e2d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.azure/docker-build.yml b/.azure/docker-build.yml
@@ -40,8 +40,8 @@ jobs:
       #maxParallel: "3"
       matrix:
         # CUDA 12.1
-        "cuda 12.1 | torch 2.3 | cudnn FE v1.5.2":
-          { CUDA_VERSION: "12.1.1", TORCH_VERSION: "2.3.0", TRITON_VERSION: "2.3.0", CUDNN_FRONTEND_VERSION: "1.5.2" }
+        "cuda 12.1 | torch 2.3.1 | cudnn FE v1.5.2":
+          { CUDA_VERSION: "12.1.1", TORCH_VERSION: "2.3.1", TRITON_VERSION: "2.3.1", CUDNN_FRONTEND_VERSION: "1.5.2" }
         "cuda 12.1 | torch 2.4 /nightly | cudnn FE v1.5.2":
           { CUDA_VERSION: "12.1.1", TORCH_VERSION: "main", TORCH_INSTALL: "source", CUDNN_FRONTEND_VERSION: "1.5.2" }
         #'cuda 12.1': # this version - '8.9.5.29-1+cuda12.1' for 'libcudnn8' was not found

diff --git a/.azure/notebook-runs.yml b/.azure/notebook-runs.yml
@@ -15,8 +15,8 @@ jobs:
   - job: jupyter
     strategy:
       matrix:
-        "ubuntu22.04 | cuda 12.1 | torch 2.3":
-          docker-image: "ubuntu22.04-cuda12.1.1-cudnn-fe1.5.2-py3.10-pt_2.3.0-dev"
+        "ubuntu22.04 | cuda 12.1 | torch 2.3.1":
+          docker-image: "ubuntu22.04-cuda12.1.1-cudnn-fe1.5.2-py3.10-pt_2.3.1-dev"
           CUDA_VERSION_MM: "121"
         "ubuntu22.04 | cuda 12.1 | torch-nightly":
           docker-image: "ubuntu22.04-cuda12.1.1-cudnn-fe1.5.2-py3.10-pt_main-dev"