diff --git a/Dockerfile_trtllm b/Dockerfile_trtllm index 8406021d..ecefc048 100644 --- a/Dockerfile_trtllm +++ b/Dockerfile_trtllm @@ -18,9 +18,7 @@ RUN cargo chef prepare --recipe-path recipe.json # CUDA dependent dependencies resolver stage FROM nvidia/cuda:12.6.3-cudnn-devel-ubuntu24.04 AS cuda-builder -RUN --mount=type=cache,target=/var/cache/apt,sharing=locked \ - --mount=type=cache,target=/var/lib/apt,sharing=locked \ - apt update && apt install -y \ +RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y \ build-essential \ cmake \ curl \ @@ -37,7 +35,7 @@ RUN --mount=type=cache,target=/var/cache/apt,sharing=locked \ python3-dev \ python3-setuptools \ tar \ - wget && \ + wget --no-install-recommends && \ pipx ensurepath ENV TGI_INSTALL_PREFIX=/usr/local/tgi