cuda8.0如何安装TensorFlow-gpu？

Posted 2023-03-12

【中文标题】cuda8.0如何安装TensorFlow-gpu？

【英文标题】：How to install TensorFlow-gpu with cuda8.0?

【发布时间】：2018-07-12 13:18:18

【问题描述】：

我尝试按照官网的说明安装，导致导入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 一起工作？或者有什么其他的方法？

谢谢！！

【参考方案1】：

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

pip install tensorflow-gpu==1.4.1

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

【讨论】：

为什么我安装 1.4.1，它显示错误：ImportError: libcudnn.so.6: cannot open shared object file: No such file or directory。它是否说它需要cuda 6.5？如何使用 CUDA8.0？

@sam 导入错误 (ImportError: libcudnn.so.6) 表示您需要适当的 CuDNN 库 (developer.nvidia.com/cudnn)。 CuDNN 与 CUDA 是分开的。正如答案所示，tensorflow-gpu==1.4.1 适用于 CUDA 8。安装 CuDNN 的说明可以在讨论中找到：askubuntu.com/questions/767269/…

谢谢，您从哪里得到这些信息？

github.com/tensorflow/tensorflow/releases/tag/v1.5.0 - 在 v1.5.0 的发行说明中，作者提到二进制文件现在是针对 CUDA 9 进行预构建的

【参考方案2】：

我遇到了类似的问题，直到我发现

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

检查您安装的 cuda 版本和 cudnn 版本，然后找出哪个版本的 tensorflow-gpu 与使用上述链接的版本兼容。

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

pip install tensorflow-gpu==1.2.0

它对我有用。