cuda8.0如何安装TensorFlow-gpu?

How to install TensorFlow-gpu with cuda8.0?

我尝试按照官网的说明安装,导致导入tensorflow时出现ImportError:

ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

我运行代码cat /usr/local/cuda/version.txt,显示我的cuda版本是8.0.61.

好像是tensorflow在找cuda 9.0。我无法升级 cuda,因为我在共享 gpu-server 上工作,而且我没有 root 权限。

有什么办法可以让tensorflow在cuda 8.0上运行吗?或者还有其他可用的方式吗?

谢谢!!

您需要为 CUDA-8 安装 1.4.1 版本

pip install tensorflow-gpu==1.4.1

最新的(版本 1.5)适用于 CUDA-9

我遇到了类似的问题,直到我找到

https://www.tensorflow.org/install/install_sources#tested_source_configurations

检查你安装的cuda版本和cudnn版本,然后找出哪个版本的tensorflow-gpu与上面提到的link兼容。

我已经安装了 cuda 8 和 cudnn v5.1,因此通过上面的检查 link tensorflow-gpu 1.2.0 是兼容的,安装后使用

pip install tensorflow-gpu==1.2.0

对我有用。