如何用cuda8.0安装TensorFlow-gpu？

Question

我试图按照官方网站上的说明安装它，导致tensorflow导致导入错误：

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一起工作？或任何其他方式？

谢谢！！

Answer 1

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

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

我已经安装了cuda 8和cudnn v5.1，因此通过检查上面的链接tensorflow-gpu 1.2.0是兼容的并且在安装之后使用

pip install tensorflow-gpu==1.2.0

它对我有用。

Answer 2

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

pip install tensorflow-gpu==1.4.1

最新版本（版本1.5）适用于CUDA-9