Dockerfile

# SPDX-FileCopyrightText: Copyright (c) 2020-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0

ARG BASE_IMAGE=nvcr.io/nvidia/tritonserver:26.06-py3
ARG TRITONSDK_BASE_IMAGE=nvcr.io/nvidia/tritonserver:26.06-py3-sdk

ARG MODEL_ANALYZER_VERSION=1.59.0dev
ARG MODEL_ANALYZER_CONTAINER_VERSION=26.10dev
FROM ${TRITONSDK_BASE_IMAGE} AS sdk

FROM ${BASE_IMAGE}
ARG MODEL_ANALYZER_VERSION
ARG MODEL_ANALYZER_CONTAINER_VERSION
ARG BASE_IMAGE
ARG TRITONSDK_BASE_IMAGE

# Ensure apt-get won't prompt for selecting options
ENV DEBIAN_FRONTEND=noninteractive
RUN apt update -qq && apt install -y docker.io wkhtmltopdf

# Install tritonclient
COPY --from=sdk /workspace/install/python /tmp/tritonclient

RUN --mount=type=secret,id=triton_ci_pip_extra_values,env=TRITON_CI_PYPI_EXTRA_VALUES \
    if [ -n "${TRITON_CI_PYPI_EXTRA_VALUES}" ]; then \
        find /tmp/tritonclient -maxdepth 1 -type f -name \
            "tritonclient-*-any*.whl" -exec pip3 install --upgrade ${TRITON_CI_PYPI_EXTRA_VALUES} {}[all] \; ; \
    else \
        find /tmp/tritonclient -maxdepth 1 -type f -name \
            "tritonclient-*-any*.whl" -exec pip3 install --upgrade {}[all] \; ; \
    fi

WORKDIR /opt/triton-model-analyzer

RUN python3 -m pip install \
        auditwheel \
        build \
        coverage \
        mkdocs \
        mkdocs-htmlproofer-plugin==0.10.3 \
        mypy \
        nvidia-pyindex \
        types-protobuf \
        types-PyYAML \
        types-requests

COPY . .

RUN python3 -m build --wheel \
    && cd dist \
    && python3 -m pip install triton*model*analyzer*.whl

RUN chmod +x /opt/triton-model-analyzer/nvidia_entrypoint.sh

ENTRYPOINT ["/opt/triton-model-analyzer/nvidia_entrypoint.sh"]

ENV MODEL_ANALYZER_VERSION=${MODEL_ANALYZER_VERSION}
ENV MODEL_ANALYZER_CONTAINER_VERSION=${MODEL_ANALYZER_CONTAINER_VERSION}
ENV TRITON_SERVER_SDK_CONTAINER_IMAGE_NAME=${TRITONSDK_BASE_IMAGE}
ENV TRITON_SERVER_CONTAINER_IMAGE_NAME=${BASE_IMAGE}