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

问题描述 投票:-1回答:2

我用了安装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安装。
python tensorflow
2个回答
0
投票

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


-2
投票

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

activate yourEnvName
conda install tensorflow
conda install tensorflow-gpu
© www.soinside.com 2019 - 2024. All rights reserved.