From 7280271d4273e154f1c0885b81973fce00febc2c Mon Sep 17 00:00:00 2001 From: Misha Chornyi Date: Fri, 14 Nov 2025 14:31:54 -0800 Subject: [PATCH] Add Model Analyzer installation to SDK container --- Dockerfile.sdk | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/Dockerfile.sdk b/Dockerfile.sdk index 1804041101..169bd73a24 100644 --- a/Dockerfile.sdk +++ b/Dockerfile.sdk @@ -219,6 +219,11 @@ RUN if [ "$TRITON_ENABLE_GPU" = "ON" ]; then \ RUN rm -f /usr/bin/python && \ ln -s /usr/bin/python3 /usr/bin/python +# Install Model Analyzer +ARG TRITON_MODEL_ANALYZER_REPO_TAG +ARG TRITON_MODEL_ANALYZER_REPO="${TRITON_REPO_ORGANIZATION}/model_analyzer@${TRITON_MODEL_ANALYZER_REPO_TAG}" +RUN pip3 install "git+${TRITON_MODEL_ANALYZER_REPO}" + # Entrypoint Banner ENV NVIDIA_PRODUCT_NAME="Triton Server SDK" COPY docker/entrypoint.d/ /opt/nvidia/entrypoint.d/