Spaces:
Running
Running
File size: 796 Bytes
4d4dd90 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
FROM nvcr.io/nvidia/pytorch:24.01-py3
LABEL description="Docker container for DUSt3R with dependencies installed. CUDA VERSION"
ENV DEVICE="cuda"
ENV MODEL="DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth"
ARG DEBIAN_FRONTEND=noninteractive
RUN apt-get update && apt-get install -y \
git=1:2.34.1-1ubuntu1.10 \
libglib2.0-0=2.72.4-0ubuntu2.2 \
&& apt-get clean \
&& rm -rf /var/lib/apt/lists/*
RUN git clone --recursive https://github.com/naver/dust3r /dust3r
WORKDIR /dust3r
RUN pip install -r requirements.txt
RUN pip install -r requirements_optional.txt
RUN pip install opencv-python==4.8.0.74
WORKDIR /dust3r/croco/models/curope/
RUN python setup.py build_ext --inplace
WORKDIR /dust3r
COPY entrypoint.sh /entrypoint.sh
RUN chmod +x /entrypoint.sh
ENTRYPOINT ["/entrypoint.sh"]
|