1. 程式人生 > >多個版本cuda呼叫

多個版本cuda呼叫

針對多個版本的cuda,其在CMakeLists.txt中尋找的方法為:

​find_package(CUDA REQUIRED)

include_directories(${CUDA_INCLUDE_DIRS})

link_directories(${CUDA_LIBRARIES})

但是如果需要對多個版本的cuda進行查詢時,此時可以對FindCUDA.cmake檔案進行修改,而FindCUDA.cmake儲存在路徑

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

此時只需要在一系列的註釋文之後增加

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

即可實現對多個版本的cuda進行自由的管理。​