python - 如何在 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:没有名为“tensorflow_hub”的模块错误，请尝试以下方法:

• 尝试安装/升级最新版本的 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'和 How to use Tensorflow Hub Model?