| 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"] |
|
|
