CMake没有正确找到CUDA库

时间:2014-11-19 13:42:59

标签: c++ cuda makefile cmake

我正在尝试构建一个需要CUDA的程序。我提供的CMake脚本:

cmake -D CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda ..

找到CUDA并且CMake正常运行:

staudt ~/workspace/clutbb/cluster/build $ cmake -D CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda ..                                                                                  
-- Found CUDA: /usr/local/cuda (found version "6.5") 
-- Found Intel TBB
-- Boost version: 1.56.0
-- Found the following Boost libraries:
--   iostreams
--   program_options
-- Looking for include file pthread.h
-- Looking for include file pthread.h - found
-- Looking for pthread_create
-- Looking for pthread_create - not found
-- Looking for pthread_create in pthreads
-- Looking for pthread_create in pthreads - not found
-- Looking for pthread_create in pthread
-- Looking for pthread_create in pthread - found
-- Found Threads: TRUE  
-- Could NOT find SDL (missing:  SDL_LIBRARY SDL_INCLUDE_DIR) 
-- Configuring done
-- Generating done
-- Build files have been written to: /home/i11/staudt/workspace/clutbb/cluster/build

然后链接器步骤失败:

staudt ~/workspace/clutbb/cluster/build $ make
[ 69%] Built target cluster
Linking CXX executable clu
CMakeFiles/clu.dir/clu.cpp.o: In function `initCUDA(int&, CUctx_st*&, int const&)':
clu.cpp:(.text+0x517): undefined reference to `cuInit'
clu.cpp:(.text+0x52b): undefined reference to `cuDeviceGet'
clu.cpp:(.text+0x53f): undefined reference to `cuCtxCreate_v2'
clu.cpp:(.text+0x559): undefined reference to `cuDeviceGetName'
clu.cpp:(.text+0x55e): undefined reference to `cuCtxSynchronize'
CMakeFiles/clu.dir/clu.cpp.o: In function `exitCUDA(int&, CUctx_st*&)':
clu.cpp:(.text+0x684): undefined reference to `cuCtxDestroy_v2'
CMakeFiles/clu.dir/clu.cpp.o: In function `main':
clu.cpp:(.text.startup+0x1092): undefined reference to `cuCtxDestroy_v2'
clu.cpp:(.text.startup+0x10d1): undefined reference to `cuCtxSynchronize'
clu.cpp:(.text.startup+0x10e1): undefined reference to `cuCtxSynchronize'
collect2: error: ld returned 1 exit status
make[2]: *** [bin/clu] Fehler 1
make[1]: *** [bin/CMakeFiles/clu.dir/all] Fehler 2
make: *** [all] Fehler 2

所需的库位于/usr/local/cuda/lib64/stubs/libcuda.so,但我怎样才能将其指向cmake或make?

3 个答案:

答案 0 :(得分:6)

在您现在发布的存档中,有多个项目层次结构。您在问题中发布的实际错误是在基于clutbb / cluster / bin目录中的clu.cpp编译和链接clu项目期间发生的。

在同一目录中,有一个CMakeLists.txt文件。此文件控制项目层次结构的此特定级别。

在这个特定的CMakeLists.txt文件中,有以下部分:

cuda_add_executable(clu clu.cpp)
target_link_libraries(clu ${CUDA_LIBRARY} ${TBB_LIBRARY} ${Boost_LIBRARIES} rt)
target_link_libraries(clu cluster)

尝试将上面的中间行修改为:

target_link_libraries(clu ${CUDA_LIBRARY} ${TBB_LIBRARY} ${Boost_LIBRARIES} rt cuda)

这应修复链接器命令行中缺少的-lcuda。可能仍需要在机器上为其提供libcuda.so的路径,但可能没有必要,具体取决于您的机器环境的设置方式。

答案 1 :(得分:0)

带有主机调用的C ++文件并不知道它需要链接到libcudart。您必须显式设置文件所在的文件/二进制文件的依赖关系,例如

target_link_libraries(clu ${CUDA_LIBRARIES})

以上回应略有错误。看起来libcuda.so无论出于何种原因都安装在意外的位置。您可以尝试将CMAKE_LIBRARY_PATH或/和CUDA_LIB_PATH设置为该路径。

我认为CUDA_LIB_PATH需要在cmake之外设置,例如export CUDA_LIB_PATH=/usr/local/cuda/lib64/stubs/

答案 2 :(得分:0)

在CMake 3.17+上执行此操作的正确方法是使用FindCUDAToolkit模块,如下所示:

find_module(CUDAToolkit REQUIRED)
target_link_libraries(my_target PRIVATE CUDA::cudart CUDA::cuda_driver)

当链接器找到目标时,CUDA::cuda_driver目标等效于-lcuda,否则它是指向正确库的绝对路径。您应该避免通过target_link_libraries添加系统库以确保可移植性。


如果您使用的是旧版本(CMake 3.0+),则可以使用(现在不推荐使用)FindCUDA模块来模仿3.17模块,如下所示:

find_package(CUDA REQUIRED)

# Do what the new package does
find_library(CUDA_DRIVER_LIBRARY
             NAMES cuda_driver cuda
             HINTS ${CUDA_TOOLKIT_ROOT_DIR}
                   ENV CUDA_PATH
             PATH_SUFFIXES nvidia/current lib64 lib/x64 lib)
if (NOT CUDA_DRIVER_LIBRARY)
    # Don't try any stub directories until we have exhausted all other search locations.
    find_library(CUDA_DRIVER_LIBRARY
                 NAMES cuda_driver cuda
                 HINTS ${CUDA_TOOLKIT_ROOT_DIR}
                       ENV CUDA_PATH
                 PATH_SUFFIXES lib64/stubs lib/x64/stubs lib/stubs stubs)
endif ()
mark_as_advanced(CUDA_DRIVER_LIBRARY)
##

target_include_directories(my_target PRIVATE ${CUDA_INCLUDE_DIRS})
target_link_libraries(my_target PRIVATE ${CUDA_LIBRARIES} ${CUDA_DRIVER_LIBRARY})

This is adapted from the official sources, here.