diff --git a/Dockerfile_trtllm b/Dockerfile_trtllm
index 46405adf..6d3c26d6 100644
--- a/Dockerfile_trtllm
+++ b/Dockerfile_trtllm
@@ -84,7 +84,7 @@ RUN mkdir $TGI_INSTALL_PREFIX && mkdir "$TGI_INSTALL_PREFIX/include" && mkdir "$
     CMAKE_INSTALL_PREFIX=$TGI_INSTALL_PREFIX cargo build --release
 
 FROM nvidia/cuda:12.6.1-cudnn-runtime-ubuntu22.04 AS runtime
-RUN apt update && apt install -y python3 && \
+RUN apt update && apt install -y python3-minimal python3-dev && \
     rm -rf /var/lib/{apt,dpkg,cache,log}/
 
 WORKDIR /usr/local/tgi/bin