Remove cudnn 8 installation

triton-inference-server · Jun 17, 2024 · 8cb38a8 · 8cb38a8
1 parent b9be792
commit 8cb38a8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/build.py b/build.py
@@ -1082,9 +1082,9 @@ def create_dockerfile_linux(
 """
     if "tensorrtllm" in backends:
         df += """
-# Remove TRT contents that are not needed in runtime
-RUN apt-get update && apt-get install -y libcudnn8-dev && ldconfig
 
+RUN ldconfig
+# Remove contents that are not needed in runtime
 RUN ARCH="$(uname -i)" \\
       && rm -fr ${TRT_ROOT}/bin ${TRT_ROOT}/targets/${ARCH}-linux-gnu/bin ${TRT_ROOT}/data \\
       && rm -fr  ${TRT_ROOT}/doc ${TRT_ROOT}/onnx_graphsurgeon ${TRT_ROOT}/python \\