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

Question

我已经在 Ubuntu 18.04 和 CUDA 9.0 和 CuDnn 7.4 中安装了 tensorflow-gpu。虽然它已成功安装但是当我导入它时

>>> 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.

Answer 1

安装说明 here 似乎表明您需要 CUDA 10 才能使用 tensorflow 1.13。降级到 tensorflow 1.12（或升级 CUDA）应该可以解决问题。