如何在 Azure ML 中使用 TensorFlow Hub

Question

我正在尝试在 Azure ML Studio 中使用 TensorFlow Hub

我正在使用内核Python 3.8 PT 和 TF

我安装了几个模块：

!pip install bert-for-tf2
!pip install sentencepiece
!pip install "tensorflow>=2.0.0"
!pip install --upgrade tensorflow-hub

通过 pip list，我可以看到它们已安装：

tensorflow                              2.8.0
tensorflow-estimator                    2.3.0
tensorflow-gpu                          2.3.0
tensorflow-hub                          0.12.0
tensorflow-io-gcs-filesystem            0.24.0

然而，当我尝试按照文档使用它时 (https://www.tensorflow.org/hub)

然后我得到经典：

ModuleNotFoundError: No module named 'tensorflow_hub'

Answer 1

要解决此 ModuleNotFoundError: No module named 'tensorflow_hub' 错误，请尝试以下方法：

• 尝试 installing/upgrading 最新版本的 tensorflow 和 tensorflow-hub 然后导入：

!pip install --upgrade tensorflow

!pip install --upgrade tensorflow_hub

import tensorflow as tf

import tensorflow_hub as hub

• 安装当前环境为新内核：

python3 -m ipykernel install --user --name=testenvironment

可以参考ModuleNotFoundError: No module named 'tensorflow_hub', No module named 'tensorflow_hub' and How to use Tensorflow Hub Model?