问题定位
step1 确保cuda目录加入动态链接库$LD_LIBRARY_PATH
在安装好cuda后:
#cuda-8.0请换成你自己的cuda实际安装文件夹名字
echo 'export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda- 8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}' >> ~/.bashrc
source ~/.bashrc
step2 libcublas.so.9.0: cannot open shared object file 历史原因
tensorflow版本高,需要高版本的cuda支持,而你安装的cuda版本不够。
#对tensorflow进行降级
pip install tensorflow-gpu==1.3
step3 cuda9.1下的此问题
cuda9.1版本太高,目前最新的tensorflow1.7是在cuda9.0下编译的,官方还没有支持cuda9.1。
下载别人编译的支持cuda9.1的tensorflow文件安装。
https://github.com/mind/wheels/releases/tag/tf1.6-gpu-cuda91
安装whl文件:
pip install ./xxxxxxxxxx.whl
step4 验证 tensorflow
$ python
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session() #在该步会显示电脑的显卡信息
>>> print(sess.run(hello))
Hello, TensorFlow!
>>> a = tf.constant(10)
>>> b = tf.constant(10)
>>> print(sess.run(a + b))
20
>>>