CUDA_HOME path for Tensorflow

tensorflow cuda cudnn
Ron Cohen picture Ron Cohen · Sep 6, 2017 · Viewed 38.9k times · Source

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?

Answer

Dmitry picture Dmitry · Sep 13, 2017

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

Related questions

Which TensorFlow and CUDA version combinations are compatible?

I have noticed that some newer TensorFlow versions are incompatible with older CUDA and cuDNN versions. Does an overview of the compatible versions or even a list of officially tested combinations exist? I can't find it in the TensorFlow documentation.

Read More
Best practice for upgrading CUDA and cuDNN for tensorflow

I'm currently in charge of getting tensorflow-gpu 1.8 to work on my machine. I've been using tf-gpu 1.2 until now, but due to some required features, I have to upgrade my installation. Before doing so, I wanted to check if there is …

Read More
How can I make tensorflow run on a GPU with capability 2.x?

I've successfully installed tensorflow (GPU) on Linux Ubuntu 16.04 and made some small changes in order to make it work with the new Ubuntu LTS release. However, I thought (who knows why) that my GPU met the minimum requirement of a …

Read More