mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-16 21:00:14 +00:00
8d06e5a9df
* add selectable python package build pipeline * update tensorrt version * update tensorrt version * Update Dockerfile.ubuntu_openvino * Update install_ubuntu.sh * add parameters for openvino base image * fix syntax error
51 lines
2.7 KiB
Text
51 lines
2.7 KiB
Text
ARG UBUNTU_VERSION=20.04
|
|
FROM ubuntu:${UBUNTU_VERSION}
|
|
|
|
ARG OPENVINO_VERSION=2021.4
|
|
ARG PYTHON_VERSION=3.8
|
|
|
|
ADD scripts /tmp/scripts
|
|
RUN /tmp/scripts/install_ubuntu.sh -p ${PYTHON_VERSION} -d EdgeDevice && \
|
|
/tmp/scripts/install_os_deps.sh -d EdgeDevice && \
|
|
/tmp/scripts/install_python_deps.sh -p ${PYTHON_VERSION} -d EdgeDevice
|
|
|
|
RUN apt update && apt install -y libnuma1 ocl-icd-libopencl1 && \
|
|
rm -rf /var/lib/apt/lists/* /tmp/scripts
|
|
|
|
WORKDIR /root
|
|
|
|
ENV INTEL_OPENVINO_DIR /opt/intel/openvino_${OPENVINO_VERSION}.752
|
|
ENV LD_LIBRARY_PATH $INTEL_OPENVINO_DIR/deployment_tools/inference_engine/lib/intel64:$INTEL_OPENVINO_DIR/deployment_tools/ngraph/lib:$INTEL_OPENVINO_DIR/deployment_tools/inference_engine/external/tbb/lib:/usr/local/openblas/lib:$LD_LIBRARY_PATH
|
|
ENV InferenceEngine_DIR $INTEL_OPENVINO_DIR/deployment_tools/inference_engine/share
|
|
ENV ngraph_DIR $INTEL_OPENVINO_DIR/deployment_tools/ngraph/cmake
|
|
ENV PYTHONPATH $INTEL_OPENVINO_DIR/tools:$PYTHONPATH
|
|
ENV IE_PLUGINS_PATH $INTEL_OPENVINO_DIR/deployment_tools/inference_engine/lib/intel64
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
ENV PATH /usr/local/gradle/bin:$PATH
|
|
|
|
RUN wget https://apt.repos.intel.com/openvino/2021/GPG-PUB-KEY-INTEL-OPENVINO-2021 && \
|
|
apt-key add GPG-PUB-KEY-INTEL-OPENVINO-2021 && rm GPG-PUB-KEY-INTEL-OPENVINO-2021 && \
|
|
cd /etc/apt/sources.list.d && \
|
|
echo "deb https://apt.repos.intel.com/openvino/2021 all main">intel-openvino-2021.list && \
|
|
apt update && \
|
|
apt install -y intel-openvino-dev-ubuntu18-2021.4.752 && \
|
|
cd ${INTEL_OPENVINO_DIR}/install_dependencies && ./install_openvino_dependencies.sh -y
|
|
|
|
RUN wget https://github.com/intel/compute-runtime/releases/download/19.41.14441/intel-gmmlib_19.3.2_amd64.deb && \
|
|
wget https://github.com/intel/compute-runtime/releases/download/19.41.14441/intel-igc-core_1.0.2597_amd64.deb && \
|
|
wget https://github.com/intel/compute-runtime/releases/download/19.41.14441/intel-igc-opencl_1.0.2597_amd64.deb && \
|
|
wget https://github.com/intel/compute-runtime/releases/download/19.41.14441/intel-opencl_19.41.14441_amd64.deb && \
|
|
wget https://github.com/intel/compute-runtime/releases/download/19.41.14441/intel-ocloc_19.41.14441_amd64.deb && \
|
|
sudo dpkg -i *.deb && rm -rf *.deb
|
|
|
|
RUN mkdir -p /opt/cmake/bin && \
|
|
wget https://github.com/Kitware/CMake/releases/download/v3.21.0/cmake-3.21.0-linux-x86_64.tar.gz && \
|
|
tar -xf cmake-3.21.0-linux-x86_64.tar.gz --strip 1 -C /opt/cmake && rm -rf /cmake-3.21.0-linux-x86_64.tar.gz && \
|
|
ln -sf /opt/cmake/bin/* /usr/bin
|
|
|
|
ARG BUILD_UID=1000
|
|
ARG BUILD_USER=onnxruntimedev
|
|
WORKDIR /home/$BUILD_USER
|
|
RUN adduser --gecos 'onnxruntime Build User' --disabled-password $BUILD_USER --uid $BUILD_UID
|
|
RUN adduser $BUILD_USER video
|
|
USER $BUILD_USER
|