web-dev-qa-db-fra.com

Chemin CUDA_HOME pour Tensorflow

Les instructions d'installation de Tensorflow linux disent:

Assurez-vous de créer la variable d’environnement CUDA_HOME comme décrit dans la documentation de NVIDIA.

Je ne trouve aucune mention de CUDA_HOME dans les instructions NVIDIA pour cuDNN v6 ou dans instructions d'installation de NVIDIA CUDA Toolkit . Est-ce que quelqu'un sait comment cette variable devrait être définie sur Linux?

20
Ron Cohen

Exécutez la commande suivante dans le terminal:

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

Où vous remplacez X.X par les deux premiers chiffres de votre numéro de version (vous pouvez le trouver, par exemple, via nvcc --version).

16
Dmitry