From 0dfe46dafbe04e2148d83ca235e8dd5b4e4be79f Mon Sep 17 00:00:00 2001 From: Baizhou Zhang Date: Sat, 24 Jan 2026 11:27:03 +0800 Subject: [PATCH] [Docker] Install cudnn==9.16 for cuda 13 image to avoid check error (#17668) --- docker/Dockerfile | 1 + 1 file changed, 1 insertion(+) diff --git a/docker/Dockerfile b/docker/Dockerfile index f700bc35f..812321b62 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -309,6 +309,7 @@ RUN --mount=type=cache,target=/root/.cache/pip if [ "${CUDA_VERSION%%.*}" = "12" python3 -m pip install nvidia-cudnn-cu12==9.16.0.29 --force-reinstall --no-deps ; \ elif [ "${CUDA_VERSION%%.*}" = "13" ]; then \ python3 -m pip install nvidia-nccl-cu13==2.28.3 --force-reinstall --no-deps ; \ + python3 -m pip install nvidia-cudnn-cu13==9.16.0.29 --force-reinstall --no-deps ; \ python3 -m pip install nvidia-cublas==13.1.0.3 --force-reinstall --no-deps ; \ python3 -m pip install nixl-cu13 --no-deps ; \ python3 -m pip install cuda-python==13.1.1 ; \