CUDA_HOME путь для Tensorflow

Question

Инструкции по установке Tensorflow Linux говорят:

Убедитесь, что вы создали переменную среды CUDA_HOME, как описано в документации NVIDIA.

Я не могу найти упоминания о CUDA_HOME в инструкциях NVIDIA для cuDNN v6 или в инструкциях по установке NVIDIA CUDA Toolkit. Кто-нибудь знает, как эта переменная должна быть установлена ​​в Linux?

Answer 1

Запустите следующую команду в терминале:

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

Где вы заменяете XX первыми двумя цифрами номера вашей версии (можно узнать, например, через 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-XX

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.

Он должен работать