From ae30b114b6c8c253acea9ec5b99ad2b05d440339 Mon Sep 17 00:00:00 2001 From: Misha Chornyi Date: Tue, 10 Feb 2026 12:50:36 -0800 Subject: [PATCH] Update Dockefile --- dockerfile/Dockerfile.triton.trt_llm_backend | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/dockerfile/Dockerfile.triton.trt_llm_backend b/dockerfile/Dockerfile.triton.trt_llm_backend index d5b25920..854ad8b6 100644 --- a/dockerfile/Dockerfile.triton.trt_llm_backend +++ b/dockerfile/Dockerfile.triton.trt_llm_backend @@ -145,7 +145,8 @@ RUN apt-get update -q=2 \ && apt-get purge -y "libnvinfer*" \ && pip3 uninstall -y tensorrt \ && rm -rf /var/lib/apt/lists/* \ - && pip3 install --no-cache-dir polygraphy==0.49.9 mpi4py==3.1.5 + && pip3 install --no-cache-dir polygraphy==0.49.9 \ + && curl -sSL https://raw.githubusercontent.com/triton-inference-server/TensorRT-LLM/refs/tags/v1.1.0/docker/common/install_mpi4py.sh | bash # Install NVRTC RUN [ "$(uname -m)" != "x86_64" ] && arch="sbsa" || arch="x86_64" \