无法导入TensorFlow_hub,收到"AttributeError: module 'tensorflow' has no attribute 'flags'"消息



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

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-86-5c017171c13e> in <module>
----> 1 import tensorflow_hub as hub
~Anaconda3Anaconda33envsmyPythonlibsite-packagestensorflow_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
~Anaconda3Anaconda33envsmyPythonlibsite-packagestensorflow_hubfeature_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
~Anaconda3Anaconda33envsmyPythonlibsite-packagestensorflow_hubmodule.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 
~Anaconda3Anaconda33envsmyPythonlibsite-packagestensorflow_hubnative_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
~Anaconda3Anaconda33envsmyPythonlibsite-packagestensorflow_hubcompressed_module_resolver.py in <module>
33 import tensorflow as tf
34 
---> 35 from tensorflow_hub import resolver
36 
37 
~Anaconda3Anaconda33envsmyPythonlibsite-packagestensorflow_hubresolver.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'

由于TensorFlow Hub'sTensorflow 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  

正如您在错误日志中看到的那样,张量流集线器内部尝试分配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 应该可以解决您的问题。

相关内容

最新更新