2.0.1

Lightning-AI · Oct 9, 2023 · 4a40f0e · 4a40f0e
1 parent 7d3071a
commit 4a40f0e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.azure/gpu-integrations.yml b/.azure/gpu-integrations.yml
@@ -22,8 +22,8 @@ jobs:
           torch-ver: "1.11.0"
           requires: "oldest"
         "latest":
-          docker-image: "pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime"
-          torch-ver: "2.1.0"
+          docker-image: "pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime"
+          torch-ver: "2.0.1"
     # how long to run the job before automatically cancelling
     timeoutInMinutes: "40"
     # how much time to give 'run always even if cancelled tasks' before stopping them