ci: bump CUDA tests to 11.7.1 (#1052)

Lightning-Universe · Jul 4, 2023 · 8c0298f · 8c0298f
1 parent 22954dc
commit 8c0298f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.azure/gpu-tests.yml b/.azure/gpu-tests.yml
@@ -19,12 +19,12 @@ jobs:
           PYTHON_VERSION: '3.8'
           TORCH_VERSION: '1.13.1'
           CUDA_VERSION_MM: '117'
-          runner_image: "nvidia/cuda:11.7.0-cudnn8-runtime-ubuntu20.04"
+          runner_image: "nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu20.04"
         'torch v2.0 | py3.9':
           PYTHON_VERSION: '3.9'
           TORCH_VERSION: '2.0.1'
           CUDA_VERSION_MM: '117'
-          runner_image: "nvidia/cuda:11.7.0-cudnn8-runtime-ubuntu20.04"
+          runner_image: "nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu20.04"
         'torch v2.0 | py3.10':
           PYTHON_VERSION: '3.10'
           TORCH_VERSION: '2.0.1'