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

Posted 2021-03-31

我用了安装tensorflow-gpu

sudo pip3 install tensorflow-gpu在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。

附加信息：

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

答案

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

另一答案

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

activate yourEnvName
conda install tensorflow
conda install tensorflow-gpu