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" \