Update Dockerfile

Dockerfile

@@ -1,13 +1,11 @@
-# Use the NVIDIA CUDA runtime as a base image (MATCHING PyTorch)
 FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu20.04
 
-# Set environment variables
 ENV DEBIAN_FRONTEND=noninteractive
 
-# Create a non-root user
+# Create non-root user
 RUN useradd -m -u 1000 user
 
-# Install system dependencies as root
+# Install system dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
     git \
     cmake \
@@ -19,33 +17,26 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     python3.9 \
     python3-pip \
     python3.9-dev \
-    python3.9-venv \
     && rm -rf /var/lib/apt/lists/*
 
-# Set Python 3.9 as the default python and pip versions
+# Set Python 3.9 as default
 RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.9 1
 RUN update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1
 
 # Switch to non-root user
 USER user
 
-# Create virtual environment
-RUN python -m venv /home/user/venv
-ENV VIRTUAL_ENV=/home/user/venv
-ENV PATH="$VIRTUAL_ENV/bin:$PATH"
-
-# Explicitly set CUDA and PyTorch environment variables
-ENV TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6"
-ENV FORCE_CUDA=1
-
-# Correct CUDA paths to ensure visibility
+# Set CUDA and environment variables
 ENV CUDA_HOME=/usr/local/cuda
 ENV PATH=${CUDA_HOME}/bin:${PATH}
 ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
+ENV TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6"
+ENV FORCE_CUDA=1
+ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
+ENV CUDA_VISIBLE_DEVICES=0
 
-# Set home to the user's home directory
 ENV HOME=/home/user \
-    PYTHONPATH=$HOME/app \
+    PYTHONPATH=/home/user/app \
     PYTHONUNBUFFERED=1 \
     GRADIO_ALLOW_FLAGGING=never \
     GRADIO_NUM_PORTS=1 \
@@ -53,34 +44,20 @@ ENV HOME=/home/user \
     GRADIO_THEME=huggingface \
     SYSTEM=spaces
 
-# Set working directory
 WORKDIR $HOME/app
 
-# Upgrade pip and install numpy<2 first
+# Upgrade pip and install numpy, then PyTorch
 RUN pip install --no-cache-dir --upgrade pip && \
-    pip install --no-cache-dir "numpy<2"
-
-# 🔥 Install PyTorch inside and outside virtualenv to prevent CUDA issues
-RUN pip install --no-cache-dir torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 && \
-    python -c "import torch; print('PyTorch version:', torch.__version__); print('CUDA Available:', torch.cuda.is_available()); print('Device:', torch.cuda.get_device_name(0))"
+    pip install --no-cache-dir "numpy<2" && \
+    pip install --no-cache-dir torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 && \
+    python -c "import torch; print('PyTorch version:', torch.__version__); print('CUDA Available:', torch.cuda.is_available())"
 
-# Clone the repository
+# Clone repository and install dependencies
 RUN git clone --recursive https://github.com/jnjaby/KEEP.git .
-
-# Copy application files
 COPY app.py .
 COPY requirements_HF.txt .
 
-# Install Python dependencies
-RUN pip install --no-cache-dir -r requirements_HF.txt
-RUN pip install --no-cache-dir gradio ffmpeg-python dlib-bin basicsr
-
-# Set CUDA device settings
-ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
-ENV CUDA_VISIBLE_DEVICES=0
-
-# Verify CUDA after all installs (Debugging Step)
-RUN python -c "import torch; print('Final Check - CUDA Available:', torch.cuda.is_available())"
+RUN pip install --no-cache-dir -r requirements_HF.txt && \
+    pip install --no-cache-dir gradio ffmpeg-python dlib-bin basicsr
 
-# Command to run the application
 CMD ["python", "app.py"]