Update Dockerfile

Dockerfile

@@ -47,21 +47,10 @@ 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 \
-    libcudnn8-dev && \
-    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
+    cudnn-cuda-11 \
+    && rm -rf /var/lib/apt/lists/*
+
+# Verify cuDNN installation
 RUN dpkg-query -W libcudnn8
 
 # Verify Python and pip versions
@@ -69,6 +58,7 @@ RUN python --version && pip --version
 
 
 
+
 USER user
 # Copy dependencies and install them via pip
 COPY . .