diff --git a/backends/trtllm/build.rs b/backends/trtllm/build.rs index aa613fd6..c9918e2c 100644 --- a/backends/trtllm/build.rs +++ b/backends/trtllm/build.rs @@ -7,7 +7,7 @@ use std::sync::LazyLock; const ADDITIONAL_BACKEND_LINK_LIBRARIES: [&str; 1] = ["spdlog"]; const CUDA_ARCH_LIST: Option<&str> = option_env!("CUDA_ARCH_LIST"); -const CUDA_REQUIRED_VERSION: &str = "12.6"; +const CUDA_REQUIRED_VERSION: &str = "12.8"; const MPI_REQUIRED_VERSION: &str = "4.1"; const INSTALL_PREFIX: Option<&str> = option_env!("CMAKE_INSTALL_PREFIX"); const TENSORRT_ROOT_DIR: Option<&str> = option_env!("TENSORRT_ROOT_DIR");