From ee0f159a4806a9cf8b3fadb0a01f98abf3841982 Mon Sep 17 00:00:00 2001 From: Manfei <41607353+ManfeiBai@users.noreply.github.com> Date: Fri, 4 Oct 2024 16:54:20 -0700 Subject: [PATCH] Update cuda_deps.yaml for r2.5 (#8219) --- infra/ansible/config/cuda_deps.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/infra/ansible/config/cuda_deps.yaml b/infra/ansible/config/cuda_deps.yaml index 2a6f0d33888..a9fa7e60131 100644 --- a/infra/ansible/config/cuda_deps.yaml +++ b/infra/ansible/config/cuda_deps.yaml @@ -3,7 +3,7 @@ cuda_deps: # List all libcudnn8 versions with `apt list -a libcudnn8` libcudnn: - "12.4": libcudnn-cuda-12=9.1.1.17-1 + "12.4": libcudnn9-cuda-12=9.1.1.17-1 "12.3": libcudnn9-cuda-12=9.0.0.312-1 "12.1": libcudnn8=8.9.2.26-1+cuda12.1 "12.0": libcudnn8=8.8.0.121-1+cuda12.0 @@ -11,7 +11,7 @@ cuda_deps: "11.7": libcudnn8=8.5.0.96-1+cuda11.7 "11.2": libcudnn8=8.1.1.33-1+cuda11.2 libcudnn-dev: - "12.4": libcudnn-dev-cuda-12=9.1.1.17-1 + "12.4": libcudnn9-dev-cuda-12=9.1.1.17-1 "12.3": libcudnn9-dev-cuda-12=9.0.0.312-1 "12.1": libcudnn8-dev=8.9.2.26-1+cuda12.1 "12.0": libcudnn8-dev=8.8.0.121-1+cuda12.0