如何使用 Tensorflow Hub 模型?

问题描述

目标 使用来自 TensorFlow example project 更具体地说 Tensorflow hub

的预训练模型

1.

  • 为此,我尝试使用以下命令安装 tensorflow_hubconda 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,但我不知道如何加载和使用它
  1. 使用 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

解决方法

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

相关问答

Selenium Web驱动程序和Java。元素在(x,y)点处不可单击。其...
Python-如何使用点“。” 访问字典成员?
Java 字符串是不可变的。到底是什么意思?
Java中的“ final”关键字如何工作?(我仍然可以修改对象。...
“loop:”在Java代码中。这是什么,为什么要编译?
java.lang.ClassNotFoundException:sun.jdbc.odbc.JdbcOdbc...