简体   繁体   中英

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!!

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

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.

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM