I am using pip3 install tensorflow==1.8.0
, but it doesn't have GPU support.
So I am using pip3 install tensorflow-gpu==1.8.0
, but it still raises an exception
libcudart.so.VERSION No such file.
Should I use colab
to install tensorflow
from source?
After pip3 list
:
tensorboard 1.10.0
tensorflow 1.10.0
tensorflow-hub 0.1.1
You can downgrade Tensorflow to a previous version without GPU support on Google Colab. I ran:
!pip install tensorflow==1.12.0
import tensorflow as tf
print(tf.__version__)
which initially returned
2.0.0-dev20190130
but when I returned to it after a few hours, I got the version I requested:
1.12.0
Trying to downgrade to a version with GPU support:
!pip install tensorflow-gpu==1.12.0
requires restarting the runtime and fails, as importing import tensorflow as tf
returns:
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
When the import fails you can always downgrade CUDA to version 9.0 using following commands
!wget https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb
!dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb
!apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
!apt-get update
!apt-get install cuda=9.0.176-1
You can check the version of CUDA by running:
!nvcc --version
This code now seems to fail, see the follow-up question at How to downgrade to tensorflow-gpu version 1.12 in google colab