Tensorflow linux installation instructions说：
确保按照描述创建CUDA_HOME环境变量 在NVIDIA文档中。
我在cuIDN v6的NVIDIA说明或NVIDIA CUDA Toolkit install instructions中找不到任何关于CUDA_HOME的提及。有谁知道应该如何在linux上设置这个变量？
答案 0 :(得分：16)
在终端中运行以下命令：
export CUDA_HOME=/usr/local/cuda-X.X
如果您将X.X替换为版本号的前两位数字（可以通过nvcc --version找到）。
答案 1 :(得分：-1)
在尝试安装 NVIDIA 高岭土时遇到了这个问题。正如@user2023370 所说，nvidia-cuda-toolkit 创建了一堆文件夹，我可以确认它会设置 CUDA_HOME。这避免了使用 export CUDA_HOME=/usr/local/cuda-X.X 设置变量的需要
tldr;安装 nvidia-cuda-toolkit 来设置 CUDA_HOME