Tensorflow的CUDA_主路径

Tensorflow的CUDA_主路径,tensorflow,cuda,cudnn,Tensorflow,Cuda,Cudnn,他们说: 确保按照所述创建CUDA_HOME环境变量 在英伟达文献中。 我在英伟达的CUDNN V6指令中找不到CUAUDHONE。有人知道linux上应该如何设置此变量吗?在终端中运行以下命令: export CUDA_HOME=/usr/local/cuda-X.X 将X.X替换为版本号的前两位数字(可以通过nvcc--version找到)。在尝试安装NVIDIAs时遇到此问题。正如@user2023370所说,nvidia cuda工具包创建了一组文件夹,我可以确认它会将cuda_设置为

他们说:

确保按照所述创建CUDA_HOME环境变量 在英伟达文献中。


<>我在英伟达的CUDNN V6指令中找不到CUAUDHONE。有人知道linux上应该如何设置此变量吗?

在终端中运行以下命令:

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

将X.X替换为版本号的前两位数字(可以通过
nvcc--version
找到)。

在尝试安装NVIDIAs时遇到此问题。正如@user2023370所说,nvidia cuda工具包创建了一组文件夹,我可以确认它会将cuda_设置为主页。这避免了使用export CUDA_HOME=/usr/local/CUDA-X.X设置变量的需要


tldr;安装nvidia cuda toolkit以设置cuda_HOME

如果在
/usr/local
中安装cuda,则在
.bashrc
文件中输入
export PATH=/usr/local/cuda-8.0/bin:$PATH export LD_LIBRARY\u PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY\u PATH
@Zindarod,这肯定是LD_LIBRARY\u路径步骤的一部分,这是关于CUDA_HOME的问题之前的步骤,但这并不能解决CUDA_HOME的问题。现在推荐的安装版本是9.0,这可能应该更新到CUDA-9.0。或者,如果你有最新版本的符号链接,可以使用
export-CUDA_HOME=/usr/local/CUDA
,注意Ubuntu 20.04上的
nvidia CUDA toolkit
软件包会创建一个文件夹
/usr/lib/cuda
其中包含
bin
include
等。因此在这种情况下,
cuda\u HOME
应该设置为/usr/lib/cuda吗?