Skip to content

Commit bd8e368

Browse files
authored
fix: use wheel files for installation in trtllm build (#2372) (#2375)
1 parent 0cf8041 commit bd8e368

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

container/Dockerfile.tensorrt_llm

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -506,7 +506,7 @@ RUN uv pip install --extra-index-url "${TENSORRTLLM_INDEX_URL}" "${TENSORRTLLM_P
506506
if [ "$ARCH" = "amd64" ]; then \
507507
pip install "triton==3.3.1"; \
508508
fi; \
509-
uv pip install ai-dynamo nixl --find-links /workspace/wheelhouse
509+
uv pip install /workspace/wheelhouse/ai_dynamo_runtime*cp312*.whl /workspace/wheelhouse/ai_dynamo*any.whl /workspace/wheelhouse/nixl*.whl
510510

511511
# Setup TRTLLM environment variables, same as in dev image
512512
ENV TRTLLM_USE_UCX_KVCACHE=1

0 commit comments

Comments
 (0)