Remove libnccl-dev installation in trtllm container (#1822)

deepjavalibrary · Apr 26, 2024 · 569cd6b · 569cd6b
1 parent b6b9451
commit 569cd6b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/serving/docker/tensorrt-llm.Dockerfile b/serving/docker/tensorrt-llm.Dockerfile
@@ -70,7 +70,7 @@ RUN apt-get update && apt-get install -y g++ wget unzip openmpi-bin libopenmpi-d
     apt-get clean -y && rm -rf /var/lib/apt/lists/*
 
 # Install latest CUDNN8
-RUN apt-get update && apt-get install -y libcudnn8 libnccl-dev && \
+RUN apt-get update && apt-get install -y libcudnn8 && \
     apt-get clean -y && rm -rf /var/lib/apt/lists/*
 
 # Install PyTorch