Remove previous TRT version

triton-inference-server · Oct 11, 2023 · 3df0787 · 3df0787
1 parent 02e8b65
commit 3df0787
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/build.py b/build.py
@@ -1323,6 +1323,10 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach
         df += """
 WORKDIR /workspace
 
+# Remove previous TRT installation
+RUN apt-get remove --purge -y tensorrt* libnvinfer*
+RUN pip uninstall -y tensorrt
+
 # Install new version of TRT using the script from TRT-LLM
 RUN apt-get update && apt-get install -y --no-install-recommends python-is-python3
 RUN git clone --single-branch --depth=1 -b {} https://{}:{}@gitlab-master.nvidia.com/ftp/tekit_backend.git tensorrtllm_backend