CUDA_HOME path for Tensorflow

Question

The Tensorflow linux installation instructions say:

Ensure that you create the CUDA_HOME environment variable as described in the NVIDIA documentation.

I cannot find any mention of CUDA_HOME in the NVIDIA instructions for cuDNN v6 or in the NVIDIA CUDA Toolkit install instructions. Does anyone know how this variable should be set on linux?

Solution

Run the following command in the terminal:

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

Where you replace X.X by the first two digits of your version number (can be found out e.g. via nvcc --version).