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

Question

我安装了 tensorflow-gpu 使用

sudo pip3 install tensorflow-gpu在 python3.6 上sudo pip3 install tensorflow-gpu

我使用的系统上安装了 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。

附加信息：

1. 我不想使用 virtualenv，因为我正在为整个系统安装 tensorflow，以便所有用户都可以使用它。

2. 我过去在用 bazel 编译后安装了 tensorflow，但只有我能够使用它。 其他用户不能，即使在将 cuda 路径导出到他们的配置文件之后。 所以，这次我试图让默认的 pip 安装工作。 我已经成功卸载了之前的 tensorflow 安装。

Answer 1

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

Answer 2

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

activate yourEnvName
conda install tensorflow
conda install tensorflow-gpu