Spaces:
Sleeping
Sleeping
File size: 648 Bytes
132e2af f2f112a 132e2af f2f112a bb15d97 f2f112a bb15d97 a9de562 bb15d97 132e2af |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
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"] |