如何使用Tensorflow Hub模型？

Question

目标使用 TensorFlow 示例项目中的预训练模型，更具体地说是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 添加 https://stackoverflow.com/a/43729857/10270590

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

Answer 2

第一个解决方案：

您可以尝试将tensorflow_hub直接安装到Jupyter Notebook使用的Python环境中。

!pip install --upgrade tensorflow_hub

或者您可以使用以下代码将其安装在您的环境中：

pip install --upgrade tensorflow_hub

第二种解决方案：

如果这也不起作用（这通常发生在您在 Jupyter 笔记本中工作时），则运行以下代码：

import sys 
!{sys.executable} -m pip install tensorflow_hub

这会将tensorflow_hub直接安装到Jupyter Notebook使用的Python环境中。