35 lines
1.0 KiB
Docker
35 lines
1.0 KiB
Docker
FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-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" \
|
|
SETUPTOOLS_USE_DISTUTILS=stdlib
|
|
|
|
RUN conda update conda -y
|
|
|
|
# Install libraries in the brand new image.
|
|
RUN apt-get -y update && apt-get install -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.
|
|
WORKDIR /opt/program
|
|
|
|
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 conda install -c "nvidia/label/cuda-12.1.1" cuda -y
|
|
ENV CUDA_HOME=$CONDA_PREFIX
|
|
|
|
ENV PATH=/usr/local/cuda/bin:$PATH
|
|
|
|
RUN cd GroundingDINO/ && python -m pip install .
|
|
|
|
COPY docker_test.py docker_test.py
|
|
|
|
CMD [ "python", "docker_test.py" ] |