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

时间:2019-03-05 10:08:49

标签: python tensorflow

我使用以下命令安装了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

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

其他信息:

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

  2. 我过去使用bazel编译后安装了tensorflow,但是只有我能够使用它。即使将cuda路径导出到其配置文件后,其他用户也无法执行。因此,这次我尝试使默认的pip安装起作用。我已经成功卸载了以前的tensorflow安装。

2 个答案:

答案 0 :(得分:0)

尝试安装其他版本的Tensorflow,例如1.11.0,即支持cuda 9的版本

答案 1 :(得分:-2)

要导入张量流,您的环境应该具有numpy.so检查numpy是否已安装?如果已安装,请使用以下命令安装tensorflow和tensorflow-gpu。

activate yourEnvName
conda install tensorflow
conda install tensorflow-gpu