Update Dockerfile

Dockerfile

@@ -52,10 +52,11 @@ RUN apt-get update && apt-get install -y \
     wget \
     && wget https://developer.download.nvidia.com/compute/cudnn/9.3.0/local_installers/cudnn-local-repo-ubuntu2204-9.3.0_1.0-1_amd64.deb \
     && dpkg -i cudnn-local-repo-ubuntu2204-9.3.0_1.0-1_amd64.deb \
-    && apt-key add /var/cudnn-local-repo-*/7fa2af80.pub \
+    && 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.* \
     && apt-get clean \
-    && rm -rf /var/cudnn-local-repo-* \
+    && rm -rf /var/lib/apt/lists/* \
     && rm cudnn-local-repo-ubuntu2204-9.3.0_1.0-1_amd64.deb
 
 # Switch back to the user