Tensorflow 的 CUDA_HOME 路径答案

【问题标题】：CUDA_HOME path for TensorflowTensorflow 的 CUDA_HOME 路径
【发布时间】：2018-02-14 07:23:36
【问题描述】：

确保按照说明创建 CUDA_HOME 环境变量在 NVIDIA 文档中。

我在 cuDNN v6 的 NVIDIA 说明或NVIDIA CUDA Toolkit install instructions 中找不到任何提及 CUDA_HOME 的内容。有谁知道这个变量应该如何在linux上设置？

【问题讨论】：

如果您在/usr/local 中安装了CUDA，那么在.bashrc 文件中输入export PATH=/usr/local/cuda-8.0/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH。
@Zindarod 这绝对是 LD_LIBRARY_PATH 步骤的一部分，这是关于 CUDA_HOME 问题之前的步骤，但这并没有解决关于 CUDA_HOME 的问题。

【解决方案1】：

在终端中运行以下命令：

export CUDA_HOME=/usr/local/cuda-X.X

您将 X.X 替换为版本号的前两位数字（可以通过例如nvcc --version 找到）。

【讨论】：

现在应该更新到 cuda-9.0，因为推荐的安装是 9.0 版。
如果您有指向最新版本的符号链接，请使用 export CUDA_HOME=/usr/local/cuda
我注意到 Ubuntu 20.04 上的 nvidia-cuda-toolkit 包创建了一个文件夹 /usr/lib/cuda ，其中包含 bin、include 等。所以大概在这种情况下 CUDA_HOME 应该设置为 /usr /lib/cuda?
@user2023370 ，我也有同样的疑问。你最后搞清楚了吗？
@desmond13 唉，没有。我发现 NVIDIA 没有正确记录此变量。

【解决方案2】：

在尝试安装 NVIDIA 高岭土时遇到了这个问题。正如@user2023370 所说，nvidia-cuda-toolkit 创建了一堆文件夹，我可以确认它将设置 CUDA_HOME。这避免了使用 export CUDA_HOME=/usr/local/cuda-X.X 设置变量的需要

tldr;安装 nvidia-cuda-toolkit 设置 CUDA_HOME

【讨论】：

【解决方案3】：

或最新版本的 cuda 将 export PATH=/usr/local/cuda-11.6/bin${PATH:+:${PATH}} 放在 bashrc 的末尾。

它应该可以工作

【讨论】：