Tensorflow 的 CUDA_HOME 路径

Question

Tensorflow linux安装说明说：

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

我在 cuDNN v6 的 NVIDIA 说明或 NVIDIA CUDA 工具包安装说明中找不到任何有关 CUDA_HOME 的提及。有谁知道这个变量在linux上应该如何设置？

Answer 1

在终端中运行以下命令：

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

将 X.X 替换为版本号的前两位（可以通过

nvcc --version

找到）。

Answer 2

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

export CUDA_HOME=$CONDA_PREFIX

您可以将其添加到

~/.bashrc

或

~/.bash_profile

以使此环境变量在会话之间保持不变。要将 bash 文件中的此更改应用到当前会话，请运行：

source ~/.bashrc

或

source ~/.bash_profile

通过发出

echo $CUDA_HOME

命令来验证这一点。

Answer 3

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

设置变量的需要

tldr；安装 nvidia-cuda-toolkit 以设置 CUDA_HOME

Answer 4

https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#post-installation-actions

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

应该有效

Answer 5

jetson AGX Xavier 安装 torch2trt

python3 setup.py 安装

回溯（最近一次调用最后一次）： 文件“setup.py”，第 24 行，位于 插件扩展模块 = CUDAExtension( 文件“/usr/local/lib/python3.8/dist-packages/torch/utils/cpp_extension.py”，第 1074 行，在 CUDAExtension 中 库目录 += 库路径(cuda=True) 文件“/usr/local/lib/python3.8/dist-packages/torch/utils/cpp_extension.py”，第1201行，位于library_paths中 如果（不是 os.path.exists(_join_cuda_home(lib_dir)）并且 文件“/usr/local/lib/python3.8/dist-packages/torch/utils/cpp_extension.py”，第 2407 行，位于 _join_cuda_home raise OSError('未设置 CUDA_HOME 环境变量。' OSError：未设置 CUDA_HOME 环境变量。请将其设置为您的 CUDA 安装根目录。