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