cuda9.1安装tensorflow问题

最新推荐文章于 2025-09-22 10:44:27 发布

原创

最新推荐文章于 2025-09-22 10:44:27 发布 · 4.1k 阅读

1 ·

CC 4.0 BY-SA版权

在尝试安装CUDA 9.1时遇到tensorflow无法找到libcublas.so.9.0的问题。原因是TensorFlow 1.7是为CUDA 9.0编译的。解决方法是下载第三方编译的适用于CUDA 9.1的TensorFlowwhl文件进行安装，此处提供了GitHub链接，并指出了根据Python版本选择合适的whl文件。

报错 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文件:

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

安装过程如下

验证 tensorflow步骤

$ python 
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!') 
>>> sess = tf.Session() #在该步会显示电脑的显卡信息 
>>> print(sess.run(hello)) Hello, TensorFlow! 
>>>