diff --git a/Dockerfile_trtllm b/Dockerfile_trtllm index b30c0fa3..14a74c00 100644 --- a/Dockerfile_trtllm +++ b/Dockerfile_trtllm @@ -100,7 +100,9 @@ COPY --from=mpi-builder /usr/local/mpi /usr/local/mpi ENV RUSTC_WRAPPER=sccache ENV CMAKE_INSTALL_PREFIX=$TGI_INSTALL_PREFIX -RUN export CMAKE_C_COMPILER_LAUNCHER=sccache && \ +RUN export CC=gcc-14 \ + export CXX=g++-14 \ + export CMAKE_C_COMPILER_LAUNCHER=sccache && \ export CMAKE_CXX_COMPILER_LAUNCHER=sccache && \ export CMAKE_CUDA_COMPILER_LAUNCHER=sccache && \ mkdir $TGI_INSTALL_PREFIX && mkdir "$TGI_INSTALL_PREFIX/include" && mkdir "$TGI_INSTALL_PREFIX/lib" && \