diff --git a/machine-learning/Dockerfile b/machine-learning/Dockerfile index f913fd3c78..c6dfbc0e6c 100644 --- 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/*