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!!
Hardware requirements. Note: TensorFlow binaries use AVX instructions which may not run on older CPUs. The following GPU-enabled devices are supported: NVIDIA® GPU card with CUDA® architectures 3.5, 5.0, 6.0, 7.0, 7.5, 8.0 and higher.
Step 1: Install TensorFlow We can either start by installing the correct drivers, CUDA, and cuDNN or by installing TensorFlow. In your Anaconda command prompt, you can create a new Conda environment and then install TensorFlow using pip. You can also install via Conda however you may not get the latest release.
You can uninstall tensorflow as well as install only tensorflow gpu. it should run only with the GPU rather than with the CPU by default. further specify the GPU on which you would like the notebook to run.
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.
If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!
Donate Us With