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?
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
).