在Ubuntu 18.04中导入tensorflow-gpu时遇到问题

时间:2019-03-21 06:10:14

标签: python tensorflow ubuntu-18.04 cudnn

我在tensorflow-gpu中将Ubuntu 18.04CUDA 9.0安装了CuDnn 7.4。虽然安装成功,但是在我导入时

>>> import tensorflow as tf

我收到错误消息

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

详细的error log

以前我已经安装并使用它,没有任何问题,但是一旦我重新安装了Ubuntu,就会出现此错误。

从错误看来,tensorflow正在寻找CUDA 10.0,我是否需要安装tensorfow-gpu的较低版本,当前版本是tensorflow-gpu==1.13.1

1 个答案:

答案 0 :(得分:1)

安装说明here似乎建议您为tensorflow 1.13使用CUDA 10。降级到tensorflow 1.12(或升级CUDA)应该可以解决此问题。