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

Question

我已经在Ubuntu 18.04 中安装了tensorflow-gpu 和CUDA 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。

Answer 1

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