39 lines
1.2 KiB
Plaintext
39 lines
1.2 KiB
Plaintext
FROM openmmlab/mmdeploy:ubuntu20.04-cuda11.8-mmdeploy1.3.1
|
|
|
|
ARG DEBIAN_FRONTEND=noninteractive
|
|
ENV LANG=C.UTF-8
|
|
ENV LC_ALL=C.UTF-8
|
|
WORKDIR /
|
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends feh
|
|
|
|
RUN git clone https://github.com/open-mmlab/mmdeploy.git --depth=1
|
|
RUN cd mmdeploy/; python3 tools/scripts/build_ubuntu_x64_ort.py
|
|
|
|
# Install MMPose
|
|
ENV FORCE_CUDA="1"
|
|
ENV MMCV_WITH_OPS=1
|
|
RUN pip3 install --upgrade --no-cache-dir openmim
|
|
RUN mim install mmengine
|
|
RUN mim install "mmcv>=2,<2.2.0"
|
|
RUN mim install "mmdet>=3"
|
|
RUN mim install "mmpose>=1.1.0"
|
|
# Fix an error when importing mmpose
|
|
RUN pip3 install --upgrade --no-cache-dir "numpy<2" scipy
|
|
RUN git clone --depth=1 --branch=main https://github.com/open-mmlab/mmpose.git
|
|
|
|
RUN echo 'export PYTHONPATH=/mmdeploy/build/lib:$PYTHONPATH' >> ~/.bashrc
|
|
RUN echo 'export LD_LIBRARY_PATH=/mmdeploy/../mmdeploy-dep/onnxruntime-linux-x64-1.8.1/lib/:$LD_LIBRARY_PATH' >> ~/.bashrc
|
|
|
|
# Show images
|
|
RUN apt-get update && apt-get install -y --no-install-recommends python3-tk
|
|
|
|
# Tool for fp16 conversion
|
|
RUN pip3 install --upgrade --no-cache-dir onnxconverter_common
|
|
|
|
# Fix an error when profiling
|
|
RUN pip3 install --upgrade --no-cache-dir "onnxruntime-gpu<1.17"
|
|
|
|
WORKDIR /mmdeploy/
|
|
CMD ["/bin/bash"]
|