CUDA_HOME ruta para Tensorflow

install tensorflow

Question

Las instrucciones de instalación de Tensorflow Linux dicen:

Asegúrese de crear la variable de entorno CUDA_HOME como se describe en la documentación de NVIDIA.

No puedo encontrar ninguna mención de CUDA_HOME en las instrucciones de NVIDIA para cuDNN v6 o en las instrucciones de instalación de NVIDIA CUDA Toolkit . ¿Alguien sabe cómo esta variable se debe establecer en Linux?

Answer 1

Ejecuta el siguiente comando en el terminal:

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

Donde reemplaza XX por los dos primeros dígitos de su número de versión (puede encontrarlo, por ejemplo, a través de nvcc --version ).