diff --git a/Dockerfile.sdk b/Dockerfile.sdk index c1e033cfab..12becb4c57 100644 --- a/Dockerfile.sdk +++ b/Dockerfile.sdk @@ -144,6 +144,10 @@ RUN if [ "$TARGETPLATFORM" = "linux/amd64" ]; then \ --jar-install-path /workspace/install/java-api-bindings; \ fi +RUN pip3 install hatch build \ + && cd /workspace/client/src/c++/perf_analyzer/genai-perf \ + && hatch version $(cat /workspace/TRITON_VERSION) \ + && python3 -m build --wheel --outdir /workspace/install/python ############################################################################ ## Create sdk container ############################################################################ @@ -211,6 +215,8 @@ RUN pip3 install --upgrade numpy pillow attrdict && \ "tritonclient-*linux*.whl" | xargs printf -- '%s[all]' | \ xargs pip3 install --upgrade +RUN pip3 install install/python/genai_perf-*.whl + # Install DCGM RUN if [ "$TRITON_ENABLE_GPU" = "ON" ]; then \ [ "$(uname -m)" != "x86_64" ] && arch="sbsa" || arch="x86_64" && \