Update Dockerfile

Dockerfile

@@ -47,15 +47,19 @@ 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 \
-    libcudnn8=8.9.3.* \
-    libcudnn8-dev=8.9.3.* && \
+    libcudnn8 \
+    libcudnn8-dev && \
     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/*
+# Download and install CuDNN 9.3.0 manually
+RUN CUDNN_VERSION=9.3.0 \
+    && CUDNN_TAR_FILE=cudnn-linux-x86_64-${CUDNN_VERSION}.tgz \
+    && wget https://developer.download.nvidia.com/compute/redist/cudnn/v9.3/${CUDNN_TAR_FILE} -O /tmp/${CUDNN_TAR_FILE} \
+    && tar -xzvf /tmp/${CUDNN_TAR_FILE} -C /tmp \
+    && cp /tmp/cuda/include/cudnn*.h /usr/local/cuda/include/ \
+    && cp /tmp/cuda/lib64/libcudnn* /usr/local/cuda/lib64/ \
+    && chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn* \
+    && rm -rf /tmp/${CUDNN_TAR_FILE} /tmp/cuda
 
 # Verify cuDNN version
 RUN dpkg-query -W libcudnn8
@@ -64,6 +68,7 @@ RUN dpkg-query -W libcudnn8
 RUN python --version && pip --version
 
 
+
 USER user
 # Copy dependencies and install them via pip
 COPY . .