Update Dockerfile

Dockerfile

@@ -54,12 +54,16 @@ RUN apt-get update && apt-get install -y \
     && dpkg -i cudnn-local-repo-ubuntu2204-9.3.0_1.0-1_amd64.deb \
     && cp /var/cudnn-local-repo-ubuntu2204-9.3.0/cudnn-local-*-keyring.gpg /usr/share/keyrings/ \
     && apt-get update \
-    && apt-cache madison libcudnn9 \
-    && apt-get install -y libcudnn9 \
-    && apt-get clean \
-    && rm -rf /var/lib/apt/lists/* \
     && rm cudnn-local-repo-ubuntu2204-9.3.0_1.0-1_amd64.deb
 
+# Check NVIDIA packages installed via pip
+RUN echo "Checking NVIDIA packages:" && pip list | grep nvidia
+
+# Install your requirements
+RUN pip install --upgrade pip
+RUN pip install -r requirements.txt
+RUN pip install gradio
+
 # Verify CuDNN installation
 RUN ldconfig && echo "Testing CuDNN installation..." \
     && if [ -f /usr/local/cuda/include/cudnn.h ]; then \
@@ -70,13 +74,6 @@ RUN ldconfig && echo "Testing CuDNN installation..." \
 # Switch back to the user
 USER user
 
-# Install Python dependencies
-COPY . .
-RUN pip install --upgrade pip
-RUN pip install -r requirements.txt
-RUN pip install gradio
-
-USER root
 # Configure git to trust the directory and set ownership
 RUN git config --global --add safe.directory /home/user/app
 RUN chown -R user:user /home/user/app