onnxruntime
21 строка · 1.4 Кб
1# --------------------------------------------------------------
2# Copyright (c) Microsoft Corporation. All rights reserved.
3# Licensed under the MIT License.
4# --------------------------------------------------------------
5# Dockerfile to run ONNXRuntime with source build for CPU
6
7FROM mcr.microsoft.com/cbl-mariner/base/python:3
8MAINTAINER Changming Sun "chasun@microsoft.com"
9ADD . /code
10
11RUN tdnf install -y tar ca-certificates build-essential cmake curl python3-devel python3-setuptools python3-wheel python3-pip python3-numpy python3-flatbuffers python3-packaging python3-protobuf
12# The latest cmake version in Mariner2 is 3.21, but we need 3.26+
13RUN /code/dockerfiles/scripts/install_cmake.sh
14
15# Prepare onnxruntime repository & build onnxruntime
16RUN cd /code && /bin/bash ./build.sh --allow_running_as_root --skip_submodule_sync --config Release --build_wheel --update --build --parallel --cmake_extra_defines ONNXRUNTIME_VERSION=$(cat ./VERSION_NUMBER)
17
18FROM mcr.microsoft.com/cbl-mariner/base/python:3
19COPY --from=0 /code/build/Linux/Release/dist /root
20COPY --from=0 /code/dockerfiles/LICENSE-IMAGE.txt /code/LICENSE-IMAGE.txt
21RUN tdnf install -y ca-certificates python3-setuptools python3-wheel python3-pip python3-numpy python3-flatbuffers python3-packaging python3-protobuf python3-mpmath python3-sympy && python3 -m pip install coloredlogs humanfriendly && python3 -m pip install --no-index --find-links /root onnxruntime && rm -rf /root/*.whl
22