From 23c03d6b7f8173ba5156134939309bd4464b6339 Mon Sep 17 00:00:00 2001
From: Andrew Powers-Holmes <aholmes@omnom.net>
Date: Mon, 15 Jul 2024 22:27:18 +1000
Subject: [PATCH] handle libcudnn9 fr

---
 docker/base/Dockerfile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docker/base/Dockerfile b/docker/base/Dockerfile
index 15aa6e1..348e79b 100644
--- a/docker/base/Dockerfile
+++ b/docker/base/Dockerfile
@@ -99,9 +99,9 @@ RUN --mount=type=cache,target=/var/cache/apt,sharing=locked \
     apt-get -y install --no-install-recommends \
         libcudnn8-dev=$(dpkg-query --showformat='${Version}' --show libcudnn8) \
     ; fi \
-  && if $(dpkg-query --show libcudnn9 2>&1 >/dev/null); then  \
+  && if $(dpkg-query --show libcudnn9-cuda-12 2>&1 >/dev/null); then  \
     apt-get -y install --no-install-recommends \
-        libcudnn9-dev=$(dpkg-query --showformat='${Version}' --show libcudnn9) \
+        libcudnn9-dev-cuda-12=$(dpkg-query --showformat='${Version}' --show libcudnn9-cuda-12) \
     ; fi \
   && apt-get clean