diff --git a/.azure/gpu-tests.yml b/.azure/gpu-tests.yml
index 2366fdbac..0cb4b99aa 100644
--- 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'