多cuda版本系统上的tensorflowgpu安装问题

2024-09-09 13:12:32 发布

您现在位置:Python中文网/ 问答频道 /正文

我安装了tensorflow gpu使用

sudo pip3 install tensorflow-gpu在python3.6上

我使用的系统上安装了CUDA10和CUDA9.0。你知道吗

我已经导出了cuda9.0路径,但是import tensorflow仍然提供

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

有没有办法强迫tensorflow使用CUDA9.0,因为根据官方文档,使用pip的默认预编译tensorflow只适用于CUDA9.0。你知道吗

其他信息:

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

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


Tags: installpip用户import路径gputensorflow系统
2条回答

要导入tensorflow,您的环境应该努比。所以呢是否使用导入numpy检查numpy是否已安装?如果已安装,则使用以下命令安装tensorflow和tensorflow gpu。你知道吗

activate yourEnvName
conda install tensorflow
conda install tensorflow-gpu

尝试安装不同版本的tensorflow,如1.11.0,即支持CUDA9的版本

相关问题 更多 >