From a56f0b109ff30f0a49ed471a2949c0beffdf3509 Mon Sep 17 00:00:00 2001
From: Yuanjing Xue <197832395+yuanjingx87@users.noreply.github.com>
Date: Tue, 21 Jul 2026 10:56:14 -0700
Subject: [PATCH] remove tensorrt install

Signed-off-by: Yuanjing Xue <197832395+yuanjingx87@users.noreply.github.com>
---
 docker/Dockerfile.multi       | 2 +-
 docker/common/install_base.sh | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/docker/Dockerfile.multi b/docker/Dockerfile.multi
index b25a24086af2..2b3885c6766e 100644
--- a/docker/Dockerfile.multi
+++ b/docker/Dockerfile.multi
@@ -49,7 +49,7 @@ RUN --mount=type=bind,source=docker/common,target=/opt/docker/common \
     NCCL_VER=${NCCL_VER} CUBLAS_VER=${CUBLAS_VER} \
     TORCH_INSTALL_TYPE=${TORCH_INSTALL_TYPE} \
     bash /opt/docker/common/install.sh --base --cmake --ccache --cuda_toolkit \
-         --tensorrt --polygraphy --mpi4py --pytorch
+         --polygraphy --mpi4py --pytorch
 
 # Install constraints after install.sh so cleanup() doesn't delete the file mid-RUN
 COPY constraints.txt /tmp/constraints.txt
diff --git a/docker/common/install_base.sh b/docker/common/install_base.sh
index 499edc16b7d0..d464d2e8642e 100644
--- a/docker/common/install_base.sh
+++ b/docker/common/install_base.sh
@@ -109,7 +109,7 @@ init_ubuntu() {
   pip3 install --ignore-installed "setuptools<80"
 
   echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> "${ENV}"
-  # Remove previous TRT installation
+  # Remove TRT installation
   if [[ $(apt list --installed | grep libnvinfer) ]]; then
     apt-get remove --purge -y libnvinfer*
   fi