Update Dockerfile

Dockerfile

@@ -1,3 +1,4 @@
+# Use the NVIDIA CUDA runtime as a base image for CUDA 12.3
 FROM nvidia/cuda:12.3.2-cudnn9-runtime-ubuntu22.04
 
 # Set environment variables
@@ -46,6 +47,17 @@ RUN apt-get update && apt-get install -y \
     && update-alternatives --install /usr/bin/pip pip /usr/local/bin/pip 1 \
     && apt-get clean
 
+# Install CuDNN 9.3.0
+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 \
+    && apt-get update \
+    && apt-get clean \
+    && rm -rf /var/cudnn-local-repo-* \
+    && rm cudnn-local-repo-ubuntu2204-9.3.0_1.0-1_amd64.deb
+
 # Switch back to the user
 USER user