ModuleNotFoundError：tensorflow 2.1.0 没有名为“tensorflow_core.estimator”的模块

Question

使用 tensorflow 时，出现以下错误消息

ERROR:root:Internal Python error in the inspect module.
Below is the traceback from this internal error.'

File "<frozen importlib._bootstrap>", line 1006, in _gcd_import
  File "<frozen importlib._bootstrap>", line 983, in _find_and_load
  File "<frozen importlib._bootstrap>", line 965, in _find_and_load_unlocked
ModuleNotFoundError: No module named 'tensorflow_core.estimator'

已安装的 tensorflow 相关包如下所示。我需要更新估算器的版本吗？如果是这种情况，如何安装正确版本的估算器？

Answer 1

TL;DR：通过确保tensorflow和tensorflow-estimator都在同一版本中解决了这个问题。（就我而言，我需要降级tensorflow-estimator，所以conda install tensorflow-estimator=2.1.0为我解决了）

您可能已经注意到，某些tensorflow版本在某些 GPU 上运行不佳，因此我会首先检查一些可用的版本conda search tensorflow；然后我会确保所选择的tensorflow构建能够真正识别我的 GPU（带有tf.config.list_physical_devices('GPU')）；最后，我会搜索匹配的tensorflow-estimator构建，conda search tensorflow-estimator然后才使用conda install tensorflow-estimator=<chosen version> -n <my_venv>.

但是，应该注意的是，如果您对使用 GPU 感兴趣，所有这些步骤都非常有用。如果不是这种情况，那么升级两个软件包（或降级/升级它们以使其版本匹配）可能是一种方法。

Answer 2

您需要降级tensorflow-estimator或升级tensorflow才能使版本匹配。您可以使用降级

pip install tensorflow-estimator==2.1.0