From 8c0298ff122f67c5974669daea5a0a43157775b0 Mon Sep 17 00:00:00 2001 From: Jirka Borovec <6035284+Borda@users.noreply.github.com> Date: Tue, 4 Jul 2023 11:30:05 +0200 Subject: [PATCH] ci: bump CUDA tests to 11.7.1 (#1052) --- .azure/gpu-tests.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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'