我正在尝试构建 this project ,它具有 CUDA 作为依赖项。但是cmake脚本在系统上找不到CUDA安装:
cls ~/workspace/gpucluster/cluster/build $ cmake ..
-- The C compiler identification is GNU 4.7.1
-- The CXX compiler identification is GNU 4.7.1
-- Check for working C compiler: /usr/bin/gcc
-- Check for working C compiler: /usr/bin/gcc -- works
-- Detecting C compiler ABI info
-- Detecting C compiler ABI info - done
-- Check for working CXX compiler: /usr/bin/c++
-- Check for working CXX compiler: /usr/bin/c++ -- works
-- Detecting CXX compiler ABI info
-- Detecting CXX compiler ABI info - done
CMake Error at /usr/share/cmake/Modules/FindCUDA.cmake:488 (message):
Specify CUDA_TOOLKIT_ROOT_DIR
Call Stack (most recent call first):
CMakeLists.txt:20 (find_package)
-- 配置不完整,出现错误!
我尝试将其作为环境变量添加到 .bashrc
,但没有效果:
export CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda-5.5
如何正确指定CUDA_TOOLKIT_ROOT_DIR
?
最佳答案
cmake 提到 CUDA_TOOLKIT_ROOT_DIR
作为 cmake 变量,而不是环境一。这就是为什么将它放入.bashrc 时它不起作用的原因。如果您查看 FindCUDA.cmake,它会清楚地表明:
The script will prompt the user to specify CUDA_TOOLKIT_ROOT_DIR if the prefix cannot be determined by the location of nvcc in the system path and REQUIRED is specified to find_package(). To use a different installed version of the toolkit set the environment variable CUDA_BIN_PATH before running cmake (e.g. CUDA_BIN_PATH=/usr/local/cuda1.0 instead of the default /usr/local/cuda) or set CUDA_TOOLKIT_ROOT_DIR after configuring. If you change the value of CUDA_TOOLKIT_ROOT_DIR, various components that depend on the path will be relocated.
所以把CUDA_BIN_PATH
放到.bashrc或者指定CUDA_TOOLKIT_ROOT_DIR
到cmake:
cmake -D CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda-5.5 ..
https://stackoverflow.com/questions/19980412/