Update Dockerfile

Dockerfile

@@ -25,8 +25,10 @@ ENV HOME=/home/user \
 # Set the working directory to the user's home directory
 WORKDIR $HOME/app
 
-# Install system dependencies as root
+# Switch to root to install system dependencies
 USER root
+
+# Install basic dependencies and add deadsnakes repository for Python 3.9
 RUN apt-get update && apt-get install -y \
     wget \
     curl \
@@ -35,6 +37,8 @@ RUN apt-get update && apt-get install -y \
     gnupg \
     libgl1-mesa-glx \
     libglib2.0-0 \
+    && add-apt-repository ppa:deadsnakes/ppa -y \
+    && apt-get update && apt-get install -y \
     python3.9 \
     python3.9-distutils \
     python3.9-dev \
@@ -43,7 +47,7 @@ RUN apt-get update && apt-get install -y \
     && update-alternatives --install /usr/bin/pip pip /usr/local/bin/pip 1 \
     && apt-get clean
 
-# Install CUDA 11.8 and cuDNN 9.3.0 directly from NVIDIA's repositories
+# Install CUDA 11.8 and cuDNN 9.3.0
 RUN 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 \
@@ -55,15 +59,17 @@ RUN curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu22
 # Verify CUDA and cuDNN installation
 RUN dpkg-query -W libcudnn9
 
-# Install Python dependencies
+# 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
+# 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
 
@@ -74,4 +80,4 @@ ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
 ENV CUDA_VISIBLE_DEVICES=0
 
 # Command to run your application
-CMD ["python", "app.py"]
+CMD ["python", "app.py"]