From 1c54c3d88c0e5577d4e01821e163cf4f581f012b Mon Sep 17 00:00:00 2001 From: Hugo Larcher Date: Fri, 13 Dec 2024 16:09:40 +0100 Subject: [PATCH] chore: Update devcontainer --- .devcontainer/Dockerfile_trtllm | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/.devcontainer/Dockerfile_trtllm b/.devcontainer/Dockerfile_trtllm index e396dc840f6..21b7114ce03 100644 --- a/.devcontainer/Dockerfile_trtllm +++ b/.devcontainer/Dockerfile_trtllm @@ -10,16 +10,16 @@ COPY . . RUN cargo chef prepare --recipe-path recipe.json # CUDA dependent dependencies resolver stage -FROM nvidia/cuda:12.6.1-cudnn-devel-ubuntu24.04 AS cuda-builder +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 && DEBIAN_FRONTEND=noninteractive apt install -y \ + apt update && apt install -y \ build-essential \ cmake \ curl \ - gcc \ - g++ \ + gcc-14 \ + g++-14 \ git \ git-lfs \ libssl-dev \ @@ -57,7 +57,7 @@ COPY backends/trtllm/scripts/install_tensorrt.sh /opt/install_tensorrt.sh RUN chmod +x /opt/install_tensorrt.sh && \ /opt/install_tensorrt.sh -# Prepare build Backend +# Build Backend FROM cuda-builder AS tgi-builder WORKDIR /usr/src/text-generation-inference