CODEWITHSUNDEEP
tensorflowtensorflow-serving
Ink
Ink

Reputation: 963

How to install TensorFlow-gpu with cuda8.0?

I tried to install it according to the instructions on official website, which results in an ImportError when I import tensorflow:

ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

I run the code cat /usr/local/cuda/version.txt, which shows that my cuda version is 8.0.61.

It seems that tensorflow is looking for cuda 9.0. I cannot upgrade the cuda as I am working on a shared gpu-server and I do not have the root authority.

Is there any way to make tensorflow work with cuda 8.0? Or any other way available?

Thanks!!

Upvotes: 22

Views: 32134

Answers (2)

Azam Rafique
Azam Rafique

Reputation: 145

I was facing the similar issue, until I found

https://www.tensorflow.org/install/install_sources#tested_source_configurations

check your installed cuda version and cudnn version and then find out which version of tensorflow-gpu is compatible with those using link mentioned above.

I had installed cuda 8 and cudnn v5.1, hence by checking above link tensorflow-gpu 1.2.0 was compatible and after installing that using

pip install tensorflow-gpu==1.2.0

It worked for me.

Upvotes: 13

layog
layog

Reputation: 4801

You'll need to install the version 1.4.1 for CUDA-8 as

pip install tensorflow-gpu==1.4.1

The latest (version 1.5) is for CUDA-9

Upvotes: 41

Related Questions