具有多个 cuda 版本的系统上的 tensorflow-gpu 安装问题答案

【问题标题】：tensorflow-gpu installation problem on system with multiple cuda versions具有多个 cuda 版本的系统上的 tensorflow-gpu 安装问题
【发布时间】：2019-03-05 10:08:48
【问题描述】：

我安装了 tensorflow-gpu 使用

sudo pip3 install tensorflow-gpu on python3.6

我使用的系统同时安装了 cuda 10 和 cuda 9.0。

我已经导出了 cuda 9.0 路径，但 import tensorflow 仍然给我

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

有什么办法可以强制 tensorflow 使用 cuda 9.0，因为根据官方文档，使用 pip 的默认预编译 tensorflow 仅适用于 cuda 9.0。

附加信息：

我不想使用 virtualenv，因为我正在为整个系统安装 tensorflow，以便所有用户都可以使用它。
我过去用 bazel 编译后安装了 tensorflow，但只有我能够使用它。其他用户不能，即使在将 cuda 路径导出到他们的配置文件之后。所以，我这次尝试让默认的 pip 安装工作。之前的tensorflow安装我已经卸载成功了。

【问题讨论】：

尝试安装不同版本的 tensorflow，例如 1.11.0，即支持 cuda 9 的版本
这行得通。不过，我安装了 1.12.0。谢谢！
@Vaibhavgusain 您能否发表您的评论作为答案，以便我接受？
很乐意提供帮助。 :)

标签： python tensorflow

【解决方案1】：

尝试安装不同版本的 tensorflow，例如 1.11.0，即支持 cuda 9 的版本

【讨论】：

【解决方案2】：

要导入 tensorflow，您的环境应该有 numpy。所以检查 numpy 是否安装或不使用 import numpy？如果已安装，则使用以下命令安装 tensorflow 和 tensorflow-gpu。

activate yourEnvName
conda install tensorflow
conda install tensorflow-gpu

【讨论】：