为 Linux 安装 CUDA：找不到 CUDA答案

【问题标题】：Installing CUDA for Linux: Could NOT find CUDA为 Linux 安装 CUDA：找不到 CUDA
【发布时间】：2018-08-24 21:15:53
【问题描述】：

我想在我的 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)

对这里发生的事情有什么想法吗？

【问题讨论】：

我猜你只安装了 repo 配置，而不是包本身。尝试apt-get update、apt-cache search cuda，然后使用apt-get install package-names 安装正确的CUDA 包。
linux getting started guide 有说明。
在 Ubuntu 16.04 上，这已解决：sudo apt install nvidia-cuda-dev nvidia-cuda-toolkit。

标签： ubuntu cuda

【解决方案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。

【讨论】：

只需设置 CUDA_TOOLKIT_ROOT_DIR 就足够了。例如：set(CUDA_TOOLKIT_ROOT_DIR "/usr/local/cuda")

【解决方案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

效果很好。

【讨论】：