Unverified Commit 4babb33c authored by Dmitry Tokarev's avatar Dmitry Tokarev Committed by GitHub
Browse files

fix: Dockerfile.trtllm - install triton when building from source on arm (#4579)


Signed-off-by: default avatarDmitry Tokarev <dtokarev@nvidia.com>
parent 57c701fb
...@@ -137,7 +137,7 @@ RUN if [ "$HAS_TRTLLM_CONTEXT" = "1" ]; then \ ...@@ -137,7 +137,7 @@ RUN if [ "$HAS_TRTLLM_CONTEXT" = "1" ]; then \
# Install from local wheel directory in build context # Install from local wheel directory in build context
WHEEL_FILE="$(find /trtllm_wheel -name "*.whl" | head -n 1)"; \ WHEEL_FILE="$(find /trtllm_wheel -name "*.whl" | head -n 1)"; \
if [ -n "$WHEEL_FILE" ]; then \ if [ -n "$WHEEL_FILE" ]; then \
uv pip install --no-cache "$WHEEL_FILE"; \ uv pip install --no-cache "$WHEEL_FILE" triton==3.5.0; \
else \ else \
echo "No wheel file found in /trtllm_wheel directory."; \ echo "No wheel file found in /trtllm_wheel directory."; \
exit 1; \ exit 1; \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment