Use dockerfile ARGS for TRTIS versions

Dockerfile

@@ -94,6 +94,8 @@ RUN bash -c 'if [ "$BUILD_CLIENTS_ONLY" != "1" ]; then \
 ############################################################################
 FROM ${TENSORFLOW_IMAGE} AS trtserver_build
 
+ARG TRTIS_VERSION=0.10.0dev
+ARG TRTIS_CONTAINER_VERSION=19.01dev
 ARG PYVER=3.5
 ARG BUILD_CLIENTS_ONLY=0
 
@@ -211,8 +213,8 @@ RUN (cd /opt/tensorflow && ./nvbuild.sh --python$PYVER --configonly) && \
     rm -rf /root/.cache/bazel && \
     rm -rf /tmp/*
 
-ENV TENSORRT_SERVER_VERSION 0.10.0dev
-ENV NVIDIA_TENSORRT_SERVER_VERSION 19.01dev
+ENV TENSORRT_SERVER_VERSION ${TRTIS_VERSION}
+ENV NVIDIA_TENSORRT_SERVER_VERSION ${TRTIS_CONTAINER_VERSION}
 ENV PYVER ${PYVER}
 
 COPY nvidia_entrypoint.sh /opt/tensorrtserver
@@ -223,11 +225,13 @@ ENTRYPOINT ["/opt/tensorrtserver/nvidia_entrypoint.sh"]
 ############################################################################
 FROM ${BASE_IMAGE}
 
+ARG TRTIS_VERSION=0.10.0dev
+ARG TRTIS_CONTAINER_VERSION=19.01dev
 ARG PYVER=3.5
 
-ENV TENSORRT_SERVER_VERSION 0.10.0dev
+ENV TENSORRT_SERVER_VERSION ${TRTIS_VERSION}
+ENV NVIDIA_TENSORRT_SERVER_VERSION ${TRTIS_CONTAINER_VERSION}
 LABEL com.nvidia.tensorrtserver.version="${TENSORRT_SERVER_VERSION}"
-ENV NVIDIA_TENSORRT_SERVER_VERSION 19.01dev
 
 ENV LD_LIBRARY_PATH /opt/tensorrtserver/lib:${LD_LIBRARY_PATH}
 ENV PATH /opt/tensorrtserver/bin:${PATH}