quoccuongle/grounding-dino
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: Update dockerfile

Browse Source
This commit is contained in:
kiennt
2025-08-18 08:06:51 +00:00
parent e09128f94c
commit 83c93e85ac
2 changed files with 17 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
Dockerfile
View File

@@ -1,35 +1,28 @@
FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-runtime FROM pytorch/pytorch:2.8.0-cuda12.6-cudnn9-runtime
ARG DEBIAN_FRONTEND=noninteractive ARG DEBIAN_FRONTEND=noninteractive
ENV CUDA_HOME=/usr/local/cuda \ ENV CUDA_HOME=/usr/local/cuda \
TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6+PTX" \ TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6+PTX"
SETUPTOOLS_USE_DISTUTILS=stdlib
RUN conda update conda -y # RUN conda update conda -y
# Install libraries in the brand new image. RUN apt -y update && apt install tesseract-ocr -y
RUN apt-get -y update && apt-get install -y --no-install-recommends \ RUN apt-get -y update && apt-get install libgl1 poppler-utils -y --no-install-recommends
wget \
build-essential \
git \
python3-opencv \
ca-certificates && \
rm -rf /var/lib/apt/lists/*
# Set the working directory for all the subsequent Dockerfile instructions. # Set the working directory for all the subsequent Dockerfile instructions.
WORKDIR /opt/program WORKDIR /app
RUN git clone https://github.com/IDEA-Research/GroundingDINO.git
RUN mkdir weights ; cd weights ; wget -q https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth ; cd .. RUN mkdir weights ; cd weights ; wget -q https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth ; cd ..
RUN conda install -c "nvidia/label/cuda-12.1.1" cuda -y COPY app.py /app/
ENV CUDA_HOME=$CONDA_PREFIX COPY pdf_converter.py /app/
COPY uv.lock /app/
COPY setup.py /app/
ADD groundingdino/ /app/groundingdino/
COPY pyproject.toml /app/
ENV PATH=/usr/local/cuda/bin:$PATH # 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 .
RUN cd GroundingDINO/ && python -m pip install . CMD [ "/app/.venv/bin/python", "app.py" ]
COPY docker_test.py docker_test.py
CMD [ "python", "docker_test.py" ]

2
setup.py
View File

@@ -38,7 +38,7 @@ def install_torch():
# Call the function to ensure torch is installed # Call the function to ensure torch is installed
# install_torch() # install_torch()
sys.path.insert( sys.path.insert(
0, "/home/nguyendc/cuong-dev/GroundingDINO/.venv/lib/python3.13/site-packages" 0, f"{__file__}/.venv/lib/python3.13/site-packages"
) )
import torch import torch