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/