Update Dockerfile

Dockerfile

@@ -54,10 +54,8 @@ RUN apt-get update && apt-get install -y \
     && dpkg -i cudnn-local-repo-ubuntu2204-9.3.0_1.0-1_amd64.deb \
     && cp /var/cudnn-local-repo-ubuntu2204-9.3.0/cudnn-local-*-keyring.gpg /usr/share/keyrings/ \
     && apt-get update \
-    && apt-get install -y \
-        libcudnn9=9.3.0.84-1+cuda12.3 \
-        libcudnn9-dev=9.3.0.84-1+cuda12.3 \
-        libcudnn9-samples=9.3.0.84-1+cuda12.3 \
+    && apt-cache madison libcudnn9 \
+    && apt-get install -y libcudnn9 \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/* \
     && rm cudnn-local-repo-ubuntu2204-9.3.0_1.0-1_amd64.deb