fix(ml): pin cudnn version (#23110)

pin cudnn version
2025-10-20 19:18:09 -04:00 · 2025-10-20 19:18:09 -04:00 · bcfdb2f9df
parent 23a34bee6f
commit bcfdb2f9df
1 changed files with 2 additions and 1 deletions
--- a/machine-learning/Dockerfile
+++ b/machine-learning/Dockerfile
@ -93,7 +93,8 @@ ENV LD_PRELOAD=/usr/lib/libmimalloc.so.2 \
    MACHINE_LEARNING_MODEL_ARENA=false

 RUN apt-get update && \
-    apt-get install --no-install-recommends -yqq libcudnn9-cuda-12 && \
+    # Pascal support was dropped in 9.11
+    apt-get install --no-install-recommends -yqq libcudnn9-cuda-12=9.10.2.21-1 && \
    apt-get clean && \
    rm -rf /var/lib/apt/lists/*