如何使用 Tensorflow Hub 模型？

Question

目标 使用来自 TensorFlow example project more specifically Tensorflow hub

的预训练模型

1.

• 为此，我尝试使用以下命令安装 tensorflow_hub：conda install -c conda-forge tensorflow-hub
• conda list 输出：.... tensorflow-hub 0.12.0 pyhca92ed8_0 conda-forge ....
• 到 sagemenaker EC2 实例的 anaconda 环境。
• 整个安装过程没有任何错误，但是当我尝试导入包时，它就像没有安装一样 import tensorflow_hub as hub
• 错误

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

ModuleNotFoundError: No module named 'tensorflow_hub'

• 我已经手动下载了模型并上传到ec2，但我不知道如何加载和使用它

3. 用 pip 添加 tensorflow_hub

1. Run `conda create -n venv_name` and `source activate venv_name`, where `venv_name` is the name of your virtual environment. 

2. Run `conda install pip`. This will install pip to your venv directory. 

3. Find your anaconda directory, and find the actual venv folder. It should be somewhere like `/anaconda/envs/venv_name/`. 

4. Install new packages by doing `/anaconda/envs/venv_name/bin/pip install package_name`. 

输出

(tf2p37) ubuntu@ip:~/anaconda3/envs/tf2p37/bin$ pip install --upgrade tensorflow-hub
Requirement already satisfied: tensorflow-hub in /home/ubuntu/anaconda3/envs/tf2p37/lib/python3.7/site-packages (0.12.0)
Requirement already satisfied: protobuf>=3.8.0 in /home/ubuntu/anaconda3/envs/tf2p37/lib/python3.7/site-packages (from tensorflow-hub) (3.15.2)
Requirement already satisfied: numpy>=1.12.0 in /home/ubuntu/anaconda3/envs/tf2p37/lib/python3.7/site-packages (from tensorflow-hub) (1.19.5)
Requirement already satisfied: six>=1.9 in /home/ubuntu/anaconda3/envs/tf2p37/lib/python3.7/site-packages (from protobuf>=3.8.0->tensorflow-hub) (1.15.0)

• 它在那里，但它仍然不想在我执行时被导入：import tensorflow_hub as hub

Answer 1

• 我刚从 Jupiter 笔记本安装
• pip install --upgrade tensorflow_hub
• 这并没有以某种方式覆盖所有其他文件。
• 基础环境是 SageMaker conda_tensorflow2_p36
• 您可以将其激活为 conda activate tensorflow2_p36