How to install TensorFlow-gpu with cuda8.0?

tensorflow tensorflow-serving
inkzk picture inkzk · Feb 2, 2018 · Viewed 30.4k times · Source

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

Answer

layog picture layog · Feb 2, 2018

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

Related questions

No variable to save error in Tensorflow

I am trying to save the model and then reuse it for classifying my images but unfortunately i am getting errors in restoring the model that i have saved. The code in which model has been created : # Deep Learning # ============= # # Assignment 4 # …

Read More
How to import an saved Tensorflow model train using tf.estimator and predict on input data

I have save the model using tf.estimator .method export_savedmodel as follows: export_dir="exportModel/" feature_spec = tf.feature_column.make_parse_example_spec(feature_columns) input_receiver_fn = tf.estimator.export.build_parsing_serving_input_receiver_fn(feature_…

Read More
How to export Estimator model with export_savedmodel function

are there any tutorials available about export_savedmodel ? I have gone through this article on tensorflow.org and unittest code on github.com, and still have no idea about how to construct the parameter serving_input_fn of function export_…

Read More