安装CUDA for Linux:找不到CUDA

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

我猜您只安装了存储库，而没有安装包。要安装它们，请执行以下操作：

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

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

Extra:构建您的库

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

您可以在配置库之前将它们的值导出到系统中，如下所示：

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

或者在配置文件中手动添加。

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

CUDA_NVCC_EXECUTABLE 
CUDA_INCLUDE_DIRS 
CUDA_CUDART_LIBRARY
CUDA_TOOLKIT_ROOT_DIR

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

Answer 2

在我的例子中，在Centos7机器上使用cuda9.2安装pytorch 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

效果很好。