FROM python:3.11-slim | |
LABEL description="Docker container for MASt3R with dependencies installed. CPU VERSION" | |
ENV DEVICE="cpu" | |
ENV MODEL="MASt3R_ViTLarge_BaseDecoder_512_dpt.pth" | |
ARG DEBIAN_FRONTEND=noninteractive | |
RUN apt-get update && apt-get install -y \ | |
git \ | |
libgl1-mesa-glx \ | |
libegl1-mesa \ | |
libxrandr2 \ | |
libxrandr2 \ | |
libxss1 \ | |
libxcursor1 \ | |
libxcomposite1 \ | |
libasound2 \ | |
libxi6 \ | |
libxtst6 \ | |
libglib2.0-0 \ | |
&& apt-get clean \ | |
&& rm -rf /var/lib/apt/lists/* | |
RUN git clone --recursive https://github.com/naver/mast3r /mast3r | |
WORKDIR /mast3r/dust3r | |
RUN pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu | |
RUN pip install -r requirements.txt | |
RUN pip install -r requirements_optional.txt | |
RUN pip install opencv-python==4.8.0.74 | |
WORKDIR /mast3r | |
RUN pip install -r requirements.txt | |
COPY entrypoint.sh /entrypoint.sh | |
RUN chmod +x /entrypoint.sh | |
ENTRYPOINT ["/entrypoint.sh"] | |