Tensorflow的CUDA_HOME路径

Question

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

我在cuIDN v6的NVIDIA说明或NVIDIA CUDA Toolkit安装说明中找不到任何关于CUDA_HOME的提及.有谁知道如何在linux上设置这个变量？

Answer 1

在终端中运行以下命令:

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

您将XX替换为版本号的前两位数字(可以通过例如途径找到nvcc --version).

现在应该更新到cuda-9.0,因为推荐的安装版本是9.0. (6认同)
或者如果您有最新版本的符号链接,只需使用`export CUDA_HOME =/usr/local/cuda` (4认同)
我注意到 Ubuntu 20.04 上的“nvidia-cuda-toolkit”包创建了一个文件夹“/usr/lib/cuda”，其中包含“bin”、“include”等。因此，在这种情况下，“CUDA_HOME”应该设置为 / usr/lib/cuda？ (2认同)

Answer 2

如果使用 anaconda 进行安装，则适用以下内容：

export CUDA_HOME=$CONDA_PREFIX

您可以将其添加到~/.bashrc或~/.bash_profile让此环境变量在会话之间保留。要将 bash 文件中的此更改应用到当前会话，请运行：source ~/.bashrc或source ~/.bash_profile

通过发出echo $CUDA_HOME命令来验证这一点。