Skip to content

Commit 46fe7a8

Browse files
authored
Removing unused values for TensorRT-LLM container build (#8472)
1 parent ea3c687 commit 46fe7a8

File tree

1 file changed

+3
-12
lines changed

1 file changed

+3
-12
lines changed

build.py

Lines changed: 3 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1283,10 +1283,9 @@ def create_dockerfile_linux(
12831283
pip3 uninstall -y setuptools
12841284
ENV LD_LIBRARY_PATH=/usr/local/tensorrt/lib/:/opt/tritonserver/backends/tensorrtllm:$LD_LIBRARY_PATH
12851285
1286-
# There are some ucc issues when spawning mpi processes with ompi v4.1.7a1.
1287-
# Downgrade to ompi v4.1.5rc2 to avoid the issue.
1288-
RUN rm -fr /opt/hpcx/ompi
1289-
COPY --from=nvcr.io/nvidia/tritonserver:24.02-py3-min /opt/hpcx/ompi /opt/hpcx/ompi
1286+
RUN apt-get update && apt-get install -y --no-install-recommends \
1287+
libprotobuf-dev \
1288+
&& rm -rf /var/lib/apt/lists/*
12901289
"""
12911290
with open(os.path.join(ddir, dockerfile_name), "w") as dfile:
12921291
dfile.write(df)
@@ -1465,14 +1464,6 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach
14651464
virtualenv \\
14661465
&& rm -rf /var/lib/apt/lists/*
14671466
"""
1468-
if "tensorrtllm" in backends:
1469-
df += """
1470-
# Updating the openssh-client to fix for the CVE-2024-6387. This can be removed when trtllm uses a later CUDA container(12.5 or later)
1471-
RUN apt-get update \\
1472-
&& apt-get install -y --no-install-recommends \\
1473-
openssh-client \\
1474-
&& rm -rf /var/lib/apt/lists/*
1475-
"""
14761467

14771468
if "vllm" in backends:
14781469
df += f"""

0 commit comments

Comments
 (0)