Skip to content

Commit 58b9c1a

Browse files
Update Dockerfile.triton.trt_llm_backend - mlx5
1 parent bd9c4f3 commit 58b9c1a

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

dockerfile/Dockerfile.triton.trt_llm_backend

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,10 @@ RUN pip3 install --no-cache-dir polygraphy==0.49.9 mpi4py==3.1.5 cmake==4.0.3 ni
149149
pkg-config \
150150
libzmq3-dev \
151151
libibverbs-dev \
152-
rdma-core-devel
152+
libnuma-dev \
153+
libucx-dev \
154+
rdma-core \
155+
ibverbs-providers
153156
ENV CMAKE_POLICY_VERSION_MINIMUM=3.5
154157

155158
RUN cd tensorrt_llm && \

0 commit comments

Comments
 (0)