onnxruntime

1
#-------------------------------------------------------------------------
2
# Copyright(C) 2021-2024 Intel Corporation.
3
# SPDX-License-Identifier: MIT
4
#--------------------------------------------------------------------------
5

6
ARG OPENVINO_VERSION=2024.2.0
7

8

9
# Build stage
10
FROM openvino/ubuntu22_runtime:${OPENVINO_VERSION} AS builder
11

12
ENV WORKDIR_PATH=/home/openvino
13
WORKDIR $WORKDIR_PATH
14
ENV DEBIAN_FRONTEND noninteractive
15

16
ARG DEVICE=CPU
17
ARG ONNXRUNTIME_REPO=https://github.com/microsoft/onnxruntime.git
18
ARG ONNXRUNTIME_BRANCH=main
19

20
ENV OpenVINO_DIR=${INTEL_OPENVINO_DIR}/runtime/cmake
21

22
USER root
23
RUN apt update; apt install -y git protobuf-compiler libprotobuf-dev
24
RUN git clone --recursive -b ${ONNXRUNTIME_BRANCH} ${ONNXRUNTIME_REPO}
25
RUN /bin/sh onnxruntime/dockerfiles/scripts/install_common_deps.sh
26
RUN ln -s cmake-* cmake-dir
27
RUN python3 -m pip install wheel
28
ENV PATH=${WORKDIR_PATH}/cmake-dir/bin:$PATH
29
RUN pip3 install onnx
30
RUN cd onnxruntime && ./build.sh --allow_running_as_root --config Release --update --build --parallel --use_openvino ${DEVICE} --build_shared_lib --build_wheel
31

32
#Steps to download sources
33
RUN cat /etc/apt/sources.list | sed 's/^# deb-src/deb-src/g' > ./temp; mv temp /etc/apt/sources.list
34
RUN apt update; apt install dpkg-dev
35
RUN mkdir /sources
36
WORKDIR /sources
37
RUN apt-get source cron iso-codes lsb-release powermgmt-base python-apt-common python3-apt python3-dbus python3-gi libapt-pkg6.0 libhogweed6 libnettle8
38
WORKDIR /
39
RUN tar cvf GPL_sources.tar.gz /sources
40

41
# Deploy stage
42
FROM openvino/ubuntu22_runtime:${OPENVINO_VERSION}
43

44
ENV DEBIAN_FRONTEND noninteractive
45
USER root
46
COPY --from=builder /home/openvino/onnxruntime/build/Linux/Release/dist/*.whl ./
47
COPY --from=builder /GPL_sources.tar.gz ./
48
RUN python3 -m pip install ./*.whl && rm ./*.whl
49
ARG BUILD_UID=1001
50
ARG BUILD_USER=onnxruntimedev
51
RUN adduser --uid $BUILD_UID $BUILD_USER
52
RUN usermod -a -G video,users ${BUILD_USER}
53
ENV WORKDIR_PATH /home/${BUILD_USER}
54
WORKDIR ${WORKDIR_PATH}
55

56
USER ${BUILD_USER}
57
ENV PATH=${WORKDIR_PATH}/miniconda/bin:${WORKDIR_PATH}/cmake-dir/bin:$PATH
58
ENV IE_PLUGINS_PATH=${INTEL_OPENVINO_DIR}/runtime/lib/intel64
59
ENV LD_LIBRARY_PATH=/opt/intel/opencl:${INTEL_OPENVINO_DIR}/runtime/3rdparty/tbb/lib:${IE_PLUGINS_PATH}:${LD_LIBRARY_PATH}
60