Spaces:
Running
Running
File size: 2,299 Bytes
3c7feee |
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 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
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"]
|