Skip to content

Commit 3e06e6b

Browse files
committed
Update Dockefile (#842)
1 parent cf0f053 commit 3e06e6b

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

dockerfile/Dockerfile.triton.trt_llm_backend

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -145,7 +145,8 @@ RUN apt-get update -q=2 \
145145
&& apt-get purge -y "libnvinfer*" \
146146
&& pip3 uninstall -y tensorrt \
147147
&& rm -rf /var/lib/apt/lists/* \
148-
&& pip3 install --no-cache-dir polygraphy==0.49.9 mpi4py==3.1.5
148+
&& pip3 install --no-cache-dir polygraphy==0.49.9 \
149+
&& curl -sSL https://raw.githubusercontent.com/triton-inference-server/TensorRT-LLM/refs/tags/v1.1.0/docker/common/install_mpi4py.sh | bash
149150

150151
# Install NVRTC
151152
RUN [ "$(uname -m)" != "x86_64" ] && arch="sbsa" || arch="x86_64" \

0 commit comments

Comments
 (0)