Skip to content

Commit 7cfa475

Browse files
[None][fix] Remove the wheel from intermediate docker storage (#7175)
Signed-off-by: Martin Marciniszyn Mehringer <[email protected]>
1 parent 9d345b3 commit 7cfa475

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

docker/Dockerfile.multi

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -131,11 +131,8 @@ FROM ${DEVEL_IMAGE} AS release
131131
RUN mkdir -p /root/.cache/pip
132132

133133
WORKDIR /app/tensorrt_llm
134-
COPY --from=wheel /src/tensorrt_llm/build/tensorrt_llm*.whl .
135-
RUN --mount=type=cache,target=/root/.cache/pip \
136-
pip install tensorrt_llm*.whl && \
137-
rm tensorrt_llm*.whl && \
138-
pip cache purge
134+
RUN --mount=type=cache,target=/root/.cache/pip --mount=type=bind,from=wheel,source=/src/tensorrt_llm/build,target=/tmp/wheel \
135+
pip install /tmp/wheel/tensorrt_llm*.whl
139136

140137
COPY README.md ./
141138
COPY docs docs

jenkins/current_image_tags.properties

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
#
1212
# NB: Typically, the suffix indicates the PR whose CI pipeline generated the images. In case that
1313
# images are adopted from PostMerge pipelines, the abbreviated commit hash is used instead.
14+
IMAGE_NAME=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm
1415
LLM_DOCKER_IMAGE=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm:pytorch-25.06-py3-x86_64-ubuntu24.04-trt10.11.0.33-skip-tritondevel-202508201630-pre-test
1516
LLM_SBSA_DOCKER_IMAGE=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm:pytorch-25.06-py3-aarch64-ubuntu24.04-trt10.11.0.33-skip-tritondevel-202508201630-pre-test
1617
LLM_ROCKYLINUX8_PY310_DOCKER_IMAGE=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm:cuda-12.9.1-devel-rocky8-x86_64-rocky8-py310-trt10.11.0.33-skip-tritondevel-202508201630-pre-test

0 commit comments

Comments
 (0)