From b665e2fa0aec9bab85eb6bd3e576eb94f881f4f6 Mon Sep 17 00:00:00 2001
From: Morgan Funtowicz <morgan@huggingface.co>
Date: Tue, 30 Jul 2024 08:45:20 +0000
Subject: [PATCH] look for cuda 12.5

---
 backends/trtllm/CMakeLists.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/backends/trtllm/CMakeLists.txt b/backends/trtllm/CMakeLists.txt
index a925eff2..80306059 100644
--- a/backends/trtllm/CMakeLists.txt
+++ b/backends/trtllm/CMakeLists.txt
@@ -14,7 +14,7 @@ set(TGI_TRTLLM_BACKEND_TRT_INCLUDE_DIR "${TGI_TRTLLM_BACKEND_TRT_ROOT}/include"
 set(TGI_TRTLLM_BACKEND_TRT_LIB_DIR "${TGI_TRTLLM_BACKEND_TRT_ROOT}/lib" CACHE STRING "Path where TensorRT libraries are located")
 
 # We are using nvidia-ml to query at runtime device information to enable some architecture-specific features
-find_package(CUDAToolkit REQUIRED COMPONENTS CUDA::nvml)
+find_package(CUDAToolkit 12.5 REQUIRED COMPONENTS CUDA::nvml)
 
 #### External dependencies ####
 include(cmake/fmt.cmake)