Skip to content

Commit b69c247

Browse files
committed
Make sure cmake knows how to find nvcc
1 parent 034b609 commit b69c247

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

dlib/CMakeLists.txt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -641,7 +641,8 @@ if (NOT TARGET dlib)
641641
if (DLIB_USE_CUDA)
642642
find_package(CUDAToolkit)
643643

644-
if (CUDAToolkit_FOUND)
644+
if (CUDAToolkit_FOUND AND CUDAToolkit_NVCC_EXECUTABLE)
645+
set(CMAKE_CUDA_COMPILER ${CUDAToolkit_NVCC_EXECUTABLE})
645646
enable_language(CUDA)
646647

647648
find_package(CUDNN)

0 commit comments

Comments
 (0)