如何用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

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

pip install tensorflow-gpu==1.4.1

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

为什么我安装1.4.1,它显示错误:ImportError:libcudnn.so.6:无法打开共享对象文件:没有这样的文件或目录.它说它需要cuda 6.5吗？我怎么能用CUDA8.0？ (3认同)
@sam导入错误(ImportError:libcudnn.so.6)表示您需要相应的CuDNN库(https://developer.nvidia.com/cudnn).CuDNN与CUDA分开.正如答案所示,`tensorflow-gpu == 1.4.1`适用于CUDA 8.有关安装CuDNN的说明,请参阅讨论:https://askubuntu.com/questions/767269/how-can-i-install-cudnn-on-ubuntu-16-04 (2认同)

Answer 2

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

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

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

pip install tensorflow-gpu == 1.2.0

它对我有用.