python - 导入错误 : No module named 'tensorflow.core'

Question

使用 pip3 安装 tensorflow-gpu 后，尝试import tensorflow as tf

时出现以下错误

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/dlpda/.local/lib/python3.5/site-packages/tensorflow/__init__.py", line 22, in <module>
    from tensorflow.python import pywrap_tensorflow  # pylint: disable=unused-import
  File "/home/dlpda/.local/lib/python3.5/site-packages/tensorflow/python/__init__.py", line 52, in <module>
    from tensorflow.core.framework.graph_pb2 import *
ImportError: No module named 'tensorflow.core'

虽然我能够在 virtualenv 中成功导入，但在外部却不行。

我应该如何解决这个问题才能导入 tensorflow？

Answer 1

这可能是 tensorflow 中的一个错误。在 1.13 中，我们有一个 __init__.py 包含:

__all__ = [_s for _s in dir() if not _s.startswith('_')]

# ... later ...
try:                                                                                                                                                                       
  del python
  del core

因此，当我们从模块中删除 python 和 core 时，我们不会从 __all__ 中删除它们。这可以防止像 from tensorflow import * 这样的用例。这应该在 Tensorflow 1.14 中用 this change 修复.