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

pin cudnn version
This commit is contained in:
Mert
2025-10-20 19:18:09 -04:00
committed by GitHub
parent 23a34bee6f
commit bcfdb2f9df

View File

@@ -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/*