Dockerfile

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}

RUN apt-get update && apt-get install -y --no-install-recommends \
    git ffmpeg libgl1 libglib2.0-0 \
    && rm -rf /var/lib/apt/lists/*

COPY . /app

RUN pip install --no-cache-dir -r /app/requirements.txt
RUN pip install --no-cache-dir -e /app/ImageBind

EXPOSE 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"]