Update Dockerfile

Dockerfile

@@ -46,9 +46,16 @@ RUN apt-get update && apt-get install -y \
     && update-alternatives --install /usr/bin/pip pip /usr/local/bin/pip 1 \
     && curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/3bf863cc.pub | gpg --batch --yes --dearmor -o /usr/share/keyrings/cuda-archive-keyring.gpg && \
     echo "deb [signed-by=/usr/share/keyrings/cuda-archive-keyring.gpg] https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/ /" > /etc/apt/sources.list.d/cuda.list && \
-    apt-get update && apt-get install -y libcudnn8=8.9.3.* libcudnn8-dev=8.9.3.* && \
+    apt-get update && apt-get install -y \
+    libcudnn8=8.9.3.* \
+    libcudnn8-dev=8.9.3.* && \
     rm -rf /var/lib/apt/lists/*
 
+# Install cuDNN 9.3.0 specifically
+RUN apt-get update && apt-get install -y \
+    libcudnn8=9.3.0.* \
+    libcudnn8-dev=9.3.0.* && \
+    rm -rf /var/lib/apt/lists/*
 
 # Verify cuDNN version
 RUN dpkg-query -W libcudnn8
@@ -56,6 +63,7 @@ RUN dpkg-query -W libcudnn8
 # Verify Python and pip versions
 RUN python --version && pip --version
 
+
 USER user
 # Copy dependencies and install them via pip
 COPY . .