Update Dockerfile

Dockerfile

@@ -34,33 +34,34 @@ 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 TCNN_CUDA_ARCHITECTURES=86;80;75;70;61;60
 ENV FORCE_CUDA=1
 
+# Correct CUDA paths to ensure visibility
 ENV CUDA_HOME=/usr/local/cuda
-ENV PATH=${CUDA_HOME}/bin:/home/${USER_NAME}/.local/bin:${PATH}
+ENV PATH=${CUDA_HOME}/bin:${PATH}
 ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
-ENV LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:${LIBRARY_PATH}
 
 # Set home to the user's home directory
 ENV HOME=/home/user \
-	PATH=/home/user/.local/bin:$PATH \
     PYTHONPATH=$HOME/app \
-	PYTHONUNBUFFERED=1 \
-	GRADIO_ALLOW_FLAGGING=never \
-	GRADIO_NUM_PORTS=1 \
-	GRADIO_SERVER_NAME=0.0.0.0 \
-	GRADIO_THEME=huggingface \
-	SYSTEM=spaces
+    PYTHONUNBUFFERED=1 \
+    GRADIO_ALLOW_FLAGGING=never \
+    GRADIO_NUM_PORTS=1 \
+    GRADIO_SERVER_NAME=0.0.0.0 \
+    GRADIO_THEME=huggingface \
+    SYSTEM=spaces
 
 # Set working directory
 WORKDIR $HOME/app
 
-# Upgrade pip and install numpy<2 first, then PyTorch (with correct CUDA version)
+# Upgrade pip and install numpy<2 first
 RUN pip install --no-cache-dir --upgrade pip && \
-    pip install --no-cache-dir "numpy<2" && \
-    pip install --no-cache-dir torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 && \
+    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))"
 
 # Clone the repository
@@ -72,12 +73,14 @@ 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
-RUN pip install basicsr
+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())"
+
 # Command to run the application
 CMD ["python", "app.py"]