[Cherry-pick][TRTLLM] take out cudnn (#2286) (#2288)

deepjavalibrary · Aug 6, 2024 · 6161b30 · 6161b30
1 parent b46b23a
commit 6161b30
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/serving/docker/tensorrt-llm.Dockerfile b/serving/docker/tensorrt-llm.Dockerfile
@@ -67,9 +67,6 @@ COPY partition /opt/djl/partition
 COPY distribution[s]/ ./
 RUN mv *.deb djl-serving_all.deb || true
 
-# Install CUDNN 8
-RUN apt-get update && apt-get install -y --no-install-recommends libcudnn9-cuda-12 && rm -rf /var/lib/apt/lists/*
-
 # Install OpenMPI and other deps
 ARG DEBIAN_FRONTEND=noninteractive
 RUN apt-get update && apt-get install -y g++ wget unzip openmpi-bin libopenmpi-dev libffi-dev git-lfs rapidjson-dev graphviz && \