29 lines
938 B
Docker
29 lines
938 B
Docker
FROM pytorch/pytorch:2.8.0-cuda12.6-cudnn9-runtime
|
|
ARG DEBIAN_FRONTEND=noninteractive
|
|
|
|
ENV CUDA_HOME=/usr/local/cuda \
|
|
TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6+PTX"
|
|
|
|
# RUN conda update conda -y
|
|
|
|
RUN apt -y update && apt install tesseract-ocr -y
|
|
RUN apt-get -y update && apt-get install libgl1 poppler-utils -y --no-install-recommends
|
|
|
|
# Set the working directory for all the subsequent Dockerfile instructions.
|
|
WORKDIR /app
|
|
|
|
RUN mkdir weights ; cd weights ; wget -q https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth ; cd ..
|
|
|
|
COPY app.py /app/
|
|
COPY pdf_converter.py /app/
|
|
COPY uv.lock /app/
|
|
COPY setup.py /app/
|
|
ADD groundingdino/ /app/groundingdino/
|
|
COPY pyproject.toml /app/
|
|
|
|
# RUN cd GroundingDINO/ && python -m pip install .
|
|
RUN pip install --no-cache-dir uv && uv sync --no-cache
|
|
RUN /app/.venv/bin/python -m pip install -e .
|
|
|
|
CMD [ "/app/.venv/bin/python", "app.py" ]
|