Update Dockerfile

Dockerfile

@@ -47,18 +47,17 @@ RUN apt-get update && apt-get install -y \
     && 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 \
-    cudnn-cuda-11 \
+    libcudnn8=8.9.3.0-1+cuda11.8 \
+    libcudnn8-dev=8.9.3.0-1+cuda11.8 \
+    libcudnn8-doc=8.9.3.0-1+cuda11.8 \
     && rm -rf /var/lib/apt/lists/*
 
 # Verify cuDNN installation
-RUN dpkg-query -W libcudnn9
+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 . .