Skip to content

Commit d6247e4

Browse files
Merge pull request #1605 from ashahba:ashahba/maintainer
PiperOrigin-RevId: 319883321
2 parents 1022615 + 7dba16a commit d6247e4

File tree

2 files changed

+11
-13
lines changed

2 files changed

+11
-13
lines changed

tensorflow_serving/tools/docker/Dockerfile.devel-mkl

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ FROM ubuntu:18.04 as base_build
1616
ARG TF_SERVING_VERSION_GIT_BRANCH=master
1717
ARG TF_SERVING_VERSION_GIT_COMMIT=head
1818

19-
LABEL maintainer="Karthik Vadla <karthik.vadla@intel.com>"
19+
LABEL maintainer="Clayne Robison <clayne.b.robison@intel.com>"
2020
LABEL tensorflow_serving_github_branchtag=${TF_SERVING_VERSION_GIT_BRANCH}
2121
LABEL tensorflow_serving_github_commit=${TF_SERVING_VERSION_GIT_COMMIT}
2222

@@ -47,6 +47,10 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
4747
apt-get clean && \
4848
rm -rf /var/lib/apt/lists/*
4949

50+
RUN curl -fSsL -O https://bootstrap.pypa.io/get-pip.py && \
51+
python3 get-pip.py && \
52+
rm get-pip.py
53+
5054
# Install python 3.6.
5155
RUN add-apt-repository ppa:deadsnakes/ppa && \
5256
apt-get update && apt-get install -y \
@@ -58,10 +62,6 @@ RUN add-apt-repository ppa:deadsnakes/ppa && \
5862
# Make python3.6 the default python version
5963
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.6 0
6064

61-
RUN curl -fSsL -O https://bootstrap.pypa.io/get-pip.py && \
62-
python3 get-pip.py && \
63-
rm get-pip.py
64-
6565
RUN pip3 --no-cache-dir install \
6666
future>=0.17.1 \
6767
grpcio \
@@ -91,12 +91,11 @@ RUN git clone --branch=${TF_SERVING_VERSION_GIT_BRANCH} https://github.com/tenso
9191
git remote add upstream https://github.com/tensorflow/serving.git && \
9292
if [ "${TF_SERVING_VERSION_GIT_COMMIT}" != "head" ]; then git checkout ${TF_SERVING_VERSION_GIT_COMMIT} ; fi
9393

94-
FROM base_build as binary_build
9594

95+
FROM base_build as binary_build
9696
# Build, and install TensorFlow Serving
9797
ARG TF_SERVING_BUILD_OPTIONS="--config=mkl --config=release"
9898
RUN echo "Building with build options: ${TF_SERVING_BUILD_OPTIONS}"
99-
10099
ARG TF_SERVING_BAZEL_OPTIONS=""
101100
RUN echo "Building with Bazel options: ${TF_SERVING_BAZEL_OPTIONS}"
102101

@@ -118,7 +117,8 @@ RUN bazel build --color=yes --curses=yes \
118117
tensorflow_serving/tools/pip_package:build_pip_package && \
119118
bazel-bin/tensorflow_serving/tools/pip_package/build_pip_package \
120119
/tmp/pip && \
121-
pip --no-cache-dir install --upgrade /tmp/pip/tensorflow_serving*.whl && \
120+
pip --no-cache-dir install --upgrade \
121+
/tmp/pip/tensorflow_serving_api-*.whl && \
122122
rm -rf /tmp/pip
123123

124124
# Copy MKL libraries

tensorflow_serving/tools/docker/Dockerfile.mkl

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ FROM ubuntu:18.04
2121
ARG TF_SERVING_VERSION_GIT_BRANCH=master
2222
ARG TF_SERVING_VERSION_GIT_COMMIT=head
2323

24-
LABEL maintainer="Karthik Vadla <karthik.vadla@intel.com>"
24+
LABEL maintainer="Clayne Robison <clayne.b.robison@intel.com>"
2525
LABEL tensorflow_serving_github_branchtag=${TF_SERVING_VERSION_GIT_BRANCH}
2626
LABEL tensorflow_serving_github_commit=${TF_SERVING_VERSION_GIT_COMMIT}
2727

@@ -42,7 +42,6 @@ COPY --from=build_image /usr/local/lib/libmklml_intel.so /usr/local/lib
4242
ENV LIBRARY_PATH '/usr/local/lib:$LIBRARY_PATH'
4343
ENV LD_LIBRARY_PATH '/usr/local/lib:$LD_LIBRARY_PATH'
4444

45-
4645
# Expose ports
4746
# gRPC
4847
EXPOSE 8500
@@ -57,9 +56,6 @@ RUN mkdir -p ${MODEL_BASE_PATH}
5756
# The only required piece is the model name in order to differentiate endpoints
5857
ENV MODEL_NAME=model
5958

60-
# Create a script that runs the model server so we can use environment variables
61-
# while also passing in arguments from the docker command line
62-
6359
# Setting MKL environment variables can improve performance.
6460
# https://www.tensorflow.org/guide/performance/overview
6561
# Read about Tuning MKL for the best performance
@@ -97,6 +93,8 @@ ENV MKLDNN_VERBOSE=0
9793
ENV TENSORFLOW_INTRA_OP_PARALLELISM=2
9894
ENV TENSORFLOW_INTER_OP_PARALLELISM=2
9995

96+
# Create a script that runs the model server so we can use environment variables
97+
# while also passing in arguments from the docker command line
10098
RUN echo '#!/bin/bash \n\n\
10199
tensorflow_model_server --port=8500 --rest_api_port=8501 \
102100
--tensorflow_intra_op_parallelism=${TENSORFLOW_INTRA_OP_PARALLELISM} \

0 commit comments

Comments
 (0)