mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-16 21:00:14 +00:00
0f7c01b49b
* Use exec form of ENTRYPOINT for docker server
# Issue
The entrypoint currently uses the shell form - this prevents users from passing in any cmdline arguments... also passing a model_path in means the server only works in the envvar is set... however this is not what the error message says!
```
$ docker run -v /home/rakelkar/try/onnxzoo/style:/mnt/models -it mcr.microsoft.com/onnxruntime/server --model_path /mnt/models/model.onnx
Version: local_build
Commit ID: default
model_path must be the location of a valid file
Allowed options:
-h [ --help ] Shows a help message and exits
--log_level arg (=info) Logging level. Allowed options (case sensitive):
verbose, info, warning, error, fatal
--model_path arg Path to ONNX model
--address arg (=0.0.0.0) The base HTTP address
--http_port arg (=8001) HTTP port to listen to requests
--num_http_threads arg (=4) Number of http threads
--grpc_port arg (=50051) GRPC port to listen to requests
```
# Fix
1. remove the env var
2. use the exec form
* Update readme to use model_path arg
39 lines
1.7 KiB
Text
39 lines
1.7 KiB
Text
#-------------------------------------------------------------------------
|
|
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT License.
|
|
#--------------------------------------------------------------------------
|
|
# Official docker container for ONNX Runtime Server
|
|
# Ubuntu 16.04, CPU version, Python 3.
|
|
#--------------------------------------------------------------------------
|
|
|
|
FROM ubuntu:16.04 AS minimal
|
|
MAINTAINER Harry Yang "huayang@microsoft.com"
|
|
|
|
FROM ubuntu:16.04 AS build
|
|
ARG PYTHON_VERSION=3.5
|
|
ARG ONNXRUNTIME_REPO=https://github.com/Microsoft/onnxruntime
|
|
ARG ONNXRUNTIME_SERVER_BRANCH=master
|
|
ENV DEBIAN_FRONTEND noninteractive
|
|
|
|
RUN apt-get update && \
|
|
apt-get install -y sudo git bash
|
|
|
|
ENV PATH="/opt/cmake/bin:${PATH}"
|
|
RUN git clone --single-branch --branch ${ONNXRUNTIME_SERVER_BRANCH} --recursive ${ONNXRUNTIME_REPO} onnxruntime
|
|
RUN /onnxruntime/tools/ci_build/github/linux/docker/scripts/install_ubuntu.sh -p ${PYTHON_VERSION} && \
|
|
/onnxruntime/tools/ci_build/github/linux/docker/scripts/install_deps.sh && \
|
|
/onnxruntime/tools/ci_build/github/linux/docker/scripts/install_server_deps.sh
|
|
|
|
ENV PATH="/usr/local/go/bin:${PATH}"
|
|
|
|
WORKDIR /
|
|
RUN mkdir -p /onnxruntime/build && \
|
|
python3 /onnxruntime/tools/ci_build/build.py --build_dir /onnxruntime/build --config Release --build_server --cmake_extra_defines ONNXRUNTIME_VERSION=$(cat ./VERSION_NUMBER)
|
|
|
|
FROM minimal AS final
|
|
WORKDIR /onnxruntime/server/
|
|
COPY --from=build /onnxruntime/build/Release/onnxruntime_server /onnxruntime/server/
|
|
COPY --from=build /onnxruntime/build/Release/libonnxruntime.so.* /lib/
|
|
RUN apt-get update \
|
|
&& apt-get install -y libgomp1
|
|
ENTRYPOINT ["/onnxruntime/server/onnxruntime_server"]
|