cuda9.1安装tensorflow问题

报错  libcublas.so.9.0: cannot open shared object file

解决办法:

cuda9.1版本太高,目前最新的tensorflow1.7是在cuda9.0下编译的,官方还没有支持cuda9.1。
下载别人编译的支持cuda9.1的tensorflow文件安装。
https://github.com/mind/wheels/releases/tag/tf1.6-gpu-cuda91
安装whl文件:
cuda9.1安装tensorflow问题_第1张图片

因为我的Python 版本是2.7的,所以选择第一个,第二个第三分别对应Python3.5与3.6

安装过程如下

cuda9.1安装tensorflow问题_第2张图片

验证 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 
>>>

cuda9.1安装tensorflow问题_第3张图片

你可能感兴趣的:(tensorflow)