Fix Docker issue

Dockerfile

@@ -1,20 +1,17 @@
-FROM pytorch/pytorch:2.3.1-cuda12.1-cudnn8-runtime
+FROM pytorch/pytorch:2.3.1-cuda12.1-cudnn8-devel
 
 WORKDIR /app
+ENV YOLO_CONFIG_DIR=/tmp/Ultralytics
+ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:/opt/conda/lib:${LD_LIBRARY_PATH}
 
-# System deps
 RUN apt-get update && apt-get install -y --no-install-recommends \
     git ffmpeg libgl1 libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy project first (so editable installs work)
 COPY . /app
 
-# Install deps
 RUN pip install --no-cache-dir -r /app/requirements.txt
-
-# Install ImageBind as editable (so "import imagebind" works)
 RUN pip install --no-cache-dir -e /app/ImageBind
 
 EXPOSE 7860
-CMD ["bash", "-lc", "uvicorn webui.app:app --host 0.0.0.0 --port 7860"]
+CMD ["bash", "-lc", "python -c \"import torch; print('torch', torch.__version__, 'cuda', torch.version.cuda, 'avail', torch.cuda.is_available())\" && uvicorn webui.app:app --host 0.0.0.0 --port 7860"]

requirements.txt

@@ -1,11 +1,3 @@
-# requirements.txt
-
-# --- PyTorch CUDA 11.8 wheels ---
---extra-index-url https://download.pytorch.org/whl/cu118
-torch==2.0.0+cu118
-torchvision==0.15.1+cu118
-torchaudio==2.0.1+cu118
-
 # --- Web backend ---
 fastapi==0.111.0
 uvicorn==0.30.1
@@ -23,6 +15,5 @@ opencv-python-headless==4.10.0.84
 # --- Models / inference ---
 ultralytics==8.4.3
 open-clip-torch==2.24.0
-
 huggingface_hub>=0.24.0
 pytorchvideo==0.1.5