Skip to content

Commit 64768a7

Browse files
committed
Tprd 287 update openssh client for triton trtllm containers (#7455)
1 parent 510c7bc commit 64768a7

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

build.py

+8
Original file line numberDiff line numberDiff line change
@@ -1234,6 +1234,14 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach
12341234
virtualenv \\
12351235
&& rm -rf /var/lib/apt/lists/*
12361236
"""
1237+
if "tensorrtllm" in backends:
1238+
df += """
1239+
# 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)
1240+
RUN apt-get update \\
1241+
&& apt-get install -y --no-install-recommends \\
1242+
openssh-client \\
1243+
&& rm -rf /var/lib/apt/lists/*
1244+
"""
12371245

12381246
if "vllm" in backends:
12391247
df += """

0 commit comments

Comments
 (0)