Skip to content

Commit 212eb07

Browse files
committed
Add Model Analyzer to SDK container (#8511)
1 parent 1323e8c commit 212eb07

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

Dockerfile.sdk

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -219,6 +219,11 @@ RUN if [ "$TRITON_ENABLE_GPU" = "ON" ]; then \
219219
RUN rm -f /usr/bin/python && \
220220
ln -s /usr/bin/python3 /usr/bin/python
221221

222+
# Install Model Analyzer
223+
ARG TRITON_MODEL_ANALYZER_REPO_TAG
224+
ARG TRITON_MODEL_ANALYZER_REPO="${TRITON_REPO_ORGANIZATION}/model_analyzer@${TRITON_MODEL_ANALYZER_REPO_TAG}"
225+
RUN pip3 install "git+${TRITON_MODEL_ANALYZER_REPO}"
226+
222227
# Entrypoint Banner
223228
ENV NVIDIA_PRODUCT_NAME="Triton Server SDK"
224229
COPY docker/entrypoint.d/ /opt/nvidia/entrypoint.d/

0 commit comments

Comments
 (0)