diff --git a/ggml/src/ggml-cuda/CMakeLists.txt b/ggml/src/ggml-cuda/CMakeLists.txt index 67af1d8ccc..3520596048 100644 --- a/ggml/src/ggml-cuda/CMakeLists.txt +++ b/ggml/src/ggml-cuda/CMakeLists.txt @@ -38,6 +38,13 @@ if (CUDAToolkit_FOUND) endif() message(STATUS "Using CUDA architectures: ${CMAKE_CUDA_ARCHITECTURES}") + if(NOT DEFINED CMAKE_CUDA_COMPILER) + if(WIN32) + set(CMAKE_CUDA_COMPILER "${CUDAToolkit_BIN_DIR}/nvcc.exe") + else() + set(CMAKE_CUDA_COMPILER "${CUDAToolkit_BIN_DIR}/nvcc") + endif() + endif() enable_language(CUDA) file(GLOB GGML_HEADERS_CUDA "*.cuh")