FROM nvidia/cuda:11.7.1-devel-ubuntu22.04
# FROM python:3.10.13-bookworm

ENV LC_ALL C.UTF-8
ENV LANG C.UTF-8
ARG UID=1000
ARG UNAME=ubuntu
ARG DEBIAN_FRONTEND=noninteractive
ENV PATH="/home/$UNAME/.local/bin:$PATH"

# setup user
RUN useradd -m -u $UID $UNAME
WORKDIR /home/$UNAME
RUN chown -R $UNAME:$UNAME /home/$UNAME

# install Python
RUN apt-get update -y && \
    apt-get install -y software-properties-common build-essential git gcc curl make vim xorg-dev libxcb-shm0 libglu1-mesa-dev clang libc++-dev libc++abi-dev libsdl2-dev ninja-build libxi-dev libtbb-dev libosmesa6-dev libudev-dev autoconf libtool && \
    add-apt-repository -y ppa:deadsnakes/ppa && \
    apt-get update -y && \
    apt-get install -y python3.10 python3.10-dev python3.10-distutils
RUN curl -sS https://bootstrap.pypa.io/get-pip.py | python3.10

USER $UNAME

# install python packages
COPY --chown=$UNAME requirements.txt /tmp/requirements.txt
RUN python3.10 -m pip install --no-cache-dir --upgrade setuptools distlib pip && \
    python3.10 -m pip install --no-cache-dir -r /tmp/requirements.txt && \
    rm /tmp/requirements.txt
RUN python3.10 -m pip install git+https://github.com/facebookresearch/detectron2.git@fc9c33b1f6e5d4c37bbb46dde19af41afc1ddb2a
RUN git clone https://github.com/isl-org/Open3D.git && \
    cd Open3D/ && \
    mkdir build && \
    cd build && \
    cmake -DENABLE_HEADLESS_RENDERING=ON \
          -DBUILD_GUI=OFF \
          -DBUILD_WEBRTC=OFF \
          -DUSE_SYSTEM_GLEW=OFF \
          -DUSE_SYSTEM_GLFW=OFF \
          .. && \
    make && \
    make install-pip-package

# copy files
COPY --chown=$UNAME . /home/$UNAME/opdmulti-demo

ENV TORCH_CUDA_ARCH_LIST='8.0'
ENV FORCE_CUDA=1
# TODO: just install into /tmp instead of creating a user folder
RUN mkdir tmp && \
    cd opdmulti-demo/mask2former/modeling/pixel_decoder/ops && \
    python3.10 setup.py build install --prefix /home/$UNAME/tmp
USER root
RUN mv /home/$UNAME/tmp/local/lib/python3.10/dist-packages/MultiScaleDeformableAttention-1.0-py3.10-linux-x86_64.egg /usr/local/lib/python3.10/dist-packages/ && rm -rf /home/$UNAME/tmp
USER $UNAME
ENV PYTHONPATH="/usr/local/lib/python3.10/dist-packages/MultiScaleDeformableAttention-1.0-py3.10-linux-x86_64.egg"

WORKDIR /home/$UNAME/opdmulti-demo

CMD ["python3.10", "app.py"]