diff --git a/CMakeLists.txt b/CMakeLists.txt index 17bd9a4..d7d0415 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -91,6 +91,7 @@ IF ( (${CMAKE_C_COMPILER_ID} STREQUAL "GNU" OR ${CMAKE_C_COMPILER_ID} STREQUAL " SET (USE_CUDA ON) INCLUDE_DIRECTORIES(${CUDA_INCLUDE_DIRS}) LINK_DIRECTORIES(${CUDA_TOOLKIT_ROOT_DIR}/lib64) + LINK_DIRECTORIES(${CUDA_TOOLKIT_ROOT_DIR}/lib64/stubs) MESSAGE (STATUS "cuda include: ${CUDA_INCLUDE_DIRS}") MESSAGE (STATUS "cuda libs: ${CUDA_TOOLKIT_ROOT_DIR}/lib64")