无法导入 TensorFlow_hub，收到“AttributeError: module 'tensorflow' has no attribute 'flags'”消息

Question

我正在尝试在我的本地 jupyter 笔记本中导入 TensorFlow 集线器，但无法这样做。我创建了一个本地 conda 环境，安装了所有软件包。当前 tf 版本：Tensorflow 2.0 和本地 tf hub 版本：tensorflow-hub 0.1.1。当我运行“import tensorflow_hub as hub”代码时，出现以下错误。

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-86-5c017171c13e> in <module>
----> 1 import tensorflow_hub as hub

~\Anaconda3\Anaconda33\envs\myPython\lib\site-packages\tensorflow_hub\__init__.py in <module>
     23 from tensorflow_hub.estimator import LatestModuleExporter
     24 from tensorflow_hub.estimator import register_module_for_export
---> 25 from tensorflow_hub.feature_column import image_embedding_column
     26 from tensorflow_hub.feature_column import text_embedding_column
     27 from tensorflow_hub.image_util import get_expected_image_size

~\Anaconda3\Anaconda33\envs\myPython\lib\site-packages\tensorflow_hub\feature_column.py in <module>
     23 import tensorflow as tf
     24 from tensorflow_hub import image_util
---> 25 from tensorflow_hub import module
     26 
     27 # TODO(b/73987364): It is not possible to extend feature columns without

~\Anaconda3\Anaconda33\envs\myPython\lib\site-packages\tensorflow_hub\module.py in <module>
     21 import tensorflow as tf
     22 from tensorflow_hub import module_spec
---> 23 from tensorflow_hub import native_module
     24 from tensorflow_hub import tensor_info
     25 

~\Anaconda3\Anaconda33\envs\myPython\lib\site-packages\tensorflow_hub\native_module.py in <module>
     24 import tensorflow as tf
     25 
---> 26 from tensorflow_hub import compressed_module_resolver
     27 from tensorflow_hub import module_def_pb2
     28 from tensorflow_hub import module_impl

~\Anaconda3\Anaconda33\envs\myPython\lib\site-packages\tensorflow_hub\compressed_module_resolver.py in <module>
     33 import tensorflow as tf
     34 
---> 35 from tensorflow_hub import resolver
     36 
     37 

~\Anaconda3\Anaconda33\envs\myPython\lib\site-packages\tensorflow_hub\resolver.py in <module>
     32 from tensorflow_hub import tf_utils
     33 
---> 34 FLAGS = tf.flags.FLAGS
     35 
     36 tf.flags.DEFINE_string(

AttributeError: module 'tensorflow' has no attribute 'flags'

Answer 1

由于TensorFlow Hub's 对Tensorflow 2.0 的初始支持是从TensorFlow Hub 0.3.0 version 开始的，因此问题出在您使用的TensorFlow Hub(0.1.1) 版本中，该版本仅支持Tensorflow 1.x 版本。

使用以下命令将您的 TensorFlow Hub 升级到最新版本（截至今天为 0.8.0）。

pip install --upgrade tensorflow-hub  

正如您在错误日志中看到的那样，tensorflow hub 在内部尝试分配FLAGS = tf.flags.FLAGS。
在 Tensorlow 2.x 中，tf.flags.FLAGS 已移至 tf.compat.flags.Flag，Tensorflow Hub 0.1.1 中未解决。

现在在最新的 Tensorflow Hub(0.8.0) 中，他们使用如下标志。

from absl import flags
FLAGS = flags.FLAGS 

升级 TensorFlow Hub 应该可以解决您的问题。