Question

我想在我的Ubuntu 14.04机器上安装CUDA。因此，我从https://developer.nvidia.com/cuda-downloads下载了.deb软件包，该软件包默认在Ubuntu软件中心打开，然后安装cuda-repo-ubuntu1404。

现在，我正在尝试构建另一个依赖于CUDA的库，在这种情况下，我收到以下错误消息：

CUDA_TOOLKIT_ROOT_DIR not found or specified
-- Could NOT find CUDA (missing:  CUDA_TOOLKIT_ROOT_DIR CUDA_NVCC_EXECUTABLE CUDA_INCLUDE_DIRS CUDA_CUDART_LIBRARY)

关于这里发生了什么的任何想法？

Answer 1

我猜你只安装了repo，而不是包。为了安装它们：

sudo dpkg -i cuda-repo-ubuntu1404_6.5-14_amd64.deb 
sudo apt-get update
sudo apt-get install cuda

然后，您必须在系统上设置环境变量（这是CUDA工具包文档中详述的post-installation任务的一部分）。有很多方法，但也许最简单的方法是在~./bashrc

上写下它们

额外：构建您的图书馆

这应解决CUDA库和驱动程序安装的问题，但如果您在系统上手动构建库，可能需要做一些额外的工作;在构建库之前设置所需的路径。

您可以在配置库之前在系统上导出它们的值，例如：

export CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda
./configure

或在配置文件上手动添加它们。

您必须对您列出的路径变量执行此操作，即：

CUDA_NVCC_EXECUTABLE 
CUDA_INCLUDE_DIRS 
CUDA_CUDART_LIBRARY
CUDA_TOOLKIT_ROOT_DIR

您可以找到有关环境变量的更多信息以及如何设置它们here。

Answer 2

在我的情况下，在pytorch机器上使用cuda9.2进行Centos7 v0.4.1 v0.4.1安装，设置CUDA_TOOLKIT_ROOT_DIR无效。相反，

export CUDA_HOME=/<path>/cuda-9.2
export LD_LIBRARY_PATH=/<path>/cuda-9.2/lib64:$LD_LIBRARY_PATH

运作良好。

安装CUDA for Linux：无法找到CUDA

2 个答案: