对于Could NOT find CUDA (missing: CUDA_TOOLKIT_ROOT_DIR CUDA_INCLUDE_DIRS CUDA_CUDART_LIBRARY) 问题的解决

怀疑可能是系统找打多个版本的CUDA信息,因此在FindCUDA.cmake文件中指定了一下解决问题

​可以在CMakeLists.txt中设置

find_package(CUDA REQUIRED)

include_directories(${CUDA_INCLUDE_DIRS})

link_directories(${CUDA_LIBRARIES})

如果需要对多个版本的cuda进行查找时,此时可以对FindCUDA.cmake文件进行修改,而FindCUDA.cmake保存在路径

/usr/share/cmake-X.X/Modules/FindCUDA.cmake

此时只需要在一系列的注释文之后增加

set(CUDA_BIN_PATH /usr/local/cuda-x.X/)

 

你可能感兴趣的:(工具环境搭建)