Update Dockerfile

Dockerfile

@@ -52,12 +52,15 @@ RUN apt-get update && apt-get install -y \
     wget \
     && wget https://developer.download.nvidia.com/compute/cudnn/9.3.0/local_installers/cudnn-local-repo-ubuntu2204-9.3.0_1.0-1_amd64.deb \
     && 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/ \
+    && cp /var/cudnn-local-repo-ubuntu2204-9.3.0/cudnn-local-D9334AA3-keyring.gpg /usr/share/keyrings/ \
     && apt-get update \
     && 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
+# Check NVIDIA packages installed via pip (using || true to prevent non-zero exit)
+RUN echo "Checking NVIDIA packages:" && pip list | grep nvidia || true
+
+# Fix permissions for pip cache
+RUN mkdir -p /home/user/.cache/pip && chown -R user:user /home/user/.cache
 
 # Install your requirements
 RUN pip install --upgrade pip