@@ -29,6 +29,8 @@ ARG BASE_IMAGE=nvcr.io/nvidia/tritonserver:21.08-py3-min
29
29
30
30
ARG TRITON_CLIENT_REPO_SUBDIR=clientrepo
31
31
ARG TRITON_COMMON_REPO_TAG=main
32
+ ARG TRITON_CORE_REPO_TAG=main
33
+ ARG TRITON_BACKEND_REPO_TAG=main
32
34
ARG TRITON_THIRD_PARTY_REPO_TAG=main
33
35
ARG TRITON_MODEL_ANALYZER_REPO_TAG=main
34
36
ARG CMAKE_UBUNTU_VERSION=20.04
@@ -94,6 +96,8 @@ RUN rm -f /usr/bin/python && \
94
96
# Build the client library and examples
95
97
ARG TRITON_CLIENT_REPO_SUBDIR
96
98
ARG TRITON_COMMON_REPO_TAG
99
+ ARG TRITON_CORE_REPO_TAG
100
+ ARG TRITON_BACKEND_REPO_TAG
97
101
ARG TRITON_THIRD_PARTY_REPO_TAG
98
102
ARG TRITON_ENABLE_GPU
99
103
@@ -105,6 +109,8 @@ WORKDIR /workspace/build
105
109
RUN cmake -DCMAKE_INSTALL_PREFIX=/workspace/install \
106
110
-DTRITON_VERSION=`cat /workspace/TRITON_VERSION` \
107
111
-DTRITON_COMMON_REPO_TAG=${TRITON_COMMON_REPO_TAG} \
112
+ -DTRITON_CORE_REPO_TAG=${TRITON_CORE_REPO_TAG} \
113
+ -DTRITON_BACKEND_REPO_TAG=${TRITON_BACKEND_REPO_TAG} \
108
114
-DTRITON_THIRD_PARTY_REPO_TAG=${TRITON_THIRD_PARTY_REPO_TAG} \
109
115
-DTRITON_ENABLE_CC_HTTP=ON -DTRITON_ENABLE_CC_GRPC=ON \
110
116
-DTRITON_ENABLE_PYTHON_HTTP=ON -DTRITON_ENABLE_PYTHON_GRPC=ON \
0 commit comments