diff --git a/dockerfiles/Dockerfile.cuda b/dockerfiles/Dockerfile.cuda index 39427d414e..a132b7b559 100644 --- a/dockerfiles/Dockerfile.cuda +++ b/dockerfiles/Dockerfile.cuda @@ -9,7 +9,7 @@ FROM nvidia/cuda:10.1-cudnn7-devel MAINTAINER Vinitra Swamy "viswamy@microsoft.com" ARG ONNXRUNTIME_REPO=https://github.com/Microsoft/onnxruntime -ARG ONNXRUNTIME_SERVER_BRANCH=master +ARG ONNXRUNTIME_BRANCH=master RUN apt-get update &&\ apt-get install -y sudo git bash @@ -19,7 +19,7 @@ ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:/code/cmake-3.14.3-Linux-x86_ ENV LD_LIBRARY_PATH /opt/miniconda/lib:$LD_LIBRARY_PATH # Prepare onnxruntime repository & build onnxruntime with CUDA -RUN git clone --single-branch --branch ${ONNXRUNTIME_SERVER_BRANCH} --recursive ${ONNXRUNTIME_REPO} onnxruntime &&\ +RUN git clone --single-branch --branch ${ONNXRUNTIME_BRANCH} --recursive ${ONNXRUNTIME_REPO} onnxruntime &&\ /bin/sh onnxruntime/dockerfiles/scripts/install_common_deps.sh &&\ cp onnxruntime/docs/Privacy.md /code/Privacy.md &&\ cp onnxruntime/ThirdPartyNotices.txt /code/ThirdPartyNotices.txt &&\ diff --git a/dockerfiles/Dockerfile.ngraph b/dockerfiles/Dockerfile.ngraph index 1ab3707e60..843ed668e3 100644 --- a/dockerfiles/Dockerfile.ngraph +++ b/dockerfiles/Dockerfile.ngraph @@ -6,7 +6,7 @@ FROM ubuntu:16.04 ARG ONNXRUNTIME_REPO=https://github.com/Microsoft/onnxruntime -ARG ONNXRUNTIME_SERVER_BRANCH=master +ARG ONNXRUNTIME_BRANCH=master ENV DEBIAN_FRONTEND noninteractive @@ -18,7 +18,7 @@ ENV LD_LIBRARY_PATH=/opt/miniconda/lib:$LD_LIBRARY_PATH ENV LANG en_US.UTF-8 WORKDIR /code -RUN git clone --single-branch --branch ${ONNXRUNTIME_SERVER_BRANCH} --recursive ${ONNXRUNTIME_REPO} onnxruntime && \ +RUN git clone --single-branch --branch ${ONNXRUNTIME_BRANCH} --recursive ${ONNXRUNTIME_REPO} onnxruntime && \ cp onnxruntime/docs/Privacy.md /code/Privacy.md && \ cp onnxruntime/dockerfiles/LICENSE-IMAGE.txt /code/LICENSE-IMAGE.txt && \ cp onnxruntime/ThirdPartyNotices.txt /code/ThirdPartyNotices.txt && \ diff --git a/dockerfiles/Dockerfile.tensorrt b/dockerfiles/Dockerfile.tensorrt index a788e85bee..723b6267fe 100644 --- a/dockerfiles/Dockerfile.tensorrt +++ b/dockerfiles/Dockerfile.tensorrt @@ -9,7 +9,7 @@ FROM nvcr.io/nvidia/tensorrt:19.09-py3 MAINTAINER Vinitra Swamy "viswamy@microsoft.com" ARG ONNXRUNTIME_REPO=https://github.com/Microsoft/onnxruntime -ARG ONNXRUNTIME_SERVER_BRANCH=master +ARG ONNXRUNTIME_BRANCH=master RUN apt-get update &&\ apt-get install -y sudo git bash @@ -19,7 +19,7 @@ ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:/code/cmake-3.14.3-Linux-x86_ ENV LD_LIBRARY_PATH /opt/miniconda/lib:$LD_LIBRARY_PATH # Prepare onnxruntime repository & build onnxruntime with TensorRT -RUN git clone --single-branch --branch ${ONNXRUNTIME_SERVER_BRANCH} --recursive ${ONNXRUNTIME_REPO} onnxruntime &&\ +RUN git clone --single-branch --branch ${ONNXRUNTIME_BRANCH} --recursive ${ONNXRUNTIME_REPO} onnxruntime &&\ /bin/sh onnxruntime/dockerfiles/scripts/install_common_deps.sh &&\ cp onnxruntime/docs/Privacy.md /code/Privacy.md &&\ cp onnxruntime/dockerfiles/LICENSE-IMAGE.txt /code/LICENSE-IMAGE.txt &&\ diff --git a/dockerfiles/README.md b/dockerfiles/README.md index 1d91e0913a..083c583c6f 100644 --- a/dockerfiles/README.md +++ b/dockerfiles/README.md @@ -27,6 +27,7 @@ Use `docker pull` with any of the images and tags below to pull an image and try | OpenVino (VAD-M) | mcr.microsoft.com/azureml/onnxruntime | :v0.5.0-openvino-r1.1-vadm | :latest-openvino-vadm | | OpenVino (MYRIAD) | mcr.microsoft.com/azureml/onnxruntime | :v0.5.0-openvino-r1.1-myriad, :v1.0.0-openvino-r1.1-myriad | :latest-openvino-myriad | | OpenVino (CPU) | mcr.microsoft.com/azureml/onnxruntime | :v1.0.0-openvino-r1.1-cpu | :latest-openvino-cpu | +| nGraph | mcr.microsoft.com/azureml/onnxruntime | :v1.0.0-ngraph-v0.26.0 | :latest-ngraph | | Server | mcr.microsoft.com/onnxruntime/server | :v0.4.0, :v0.5.0, v0.5.1, v1.0.0 | :latest | ---