Skip to content

Commit 23485b8

Browse files
mc-nvdebermudez
andcommitted
Install genai-pa into SDK container (#6942)
* Update Docerkfile to install genai * Change the installation script * install both build and hatch * Update name --------- Co-authored-by: Elias Bermudez <[email protected]>
1 parent 6559cea commit 23485b8

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

Dockerfile.sdk

+6
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,10 @@ RUN if [ "$TARGETPLATFORM" = "linux/amd64" ]; then \
144144
--jar-install-path /workspace/install/java-api-bindings; \
145145
fi
146146

147+
RUN pip3 install hatch build \
148+
&& cd /workspace/client/src/c++/perf_analyzer/genai-perf \
149+
&& hatch version $(cat /workspace/TRITON_VERSION) \
150+
&& python3 -m build --wheel --outdir /workspace/install/python
147151
############################################################################
148152
## Create sdk container
149153
############################################################################
@@ -211,6 +215,8 @@ RUN pip3 install --upgrade numpy pillow attrdict && \
211215
"tritonclient-*linux*.whl" | xargs printf -- '%s[all]' | \
212216
xargs pip3 install --upgrade
213217

218+
RUN pip3 install install/python/genai_perf-*.whl
219+
214220
# Install DCGM
215221
RUN if [ "$TRITON_ENABLE_GPU" = "ON" ]; then \
216222
[ "$(uname -m)" != "x86_64" ] && arch="sbsa" || arch="x86_64" && \

0 commit comments

Comments
 (0)