安装 tensorflow_hub
您可以与 TensorFlow 1 和 TensorFlow 2 一起安装 tensorflow_hub
库。我们建议新用户立即开始使用 TensorFlow 2,现有用户则升级到 TensorFlow 2。
与 TensorFlow 2 一起使用
使用 pip 按常规方式 安装 TensorFlow 2。(有关 GPU 支持的额外说明,请参见此处。)然后,在 TensorFlow 2 旁边安装最新版本的 tensorflow-hub
(必须为 0.5.0 或更高版本)。
$ pip install "tensorflow>=2.0.0"
$ pip install --upgrade tensorflow-hub
TensorFlow Hub 的 TF1 风格 API 与 TensorFlow 2 的 v1 兼容模式一起使用。
与 TensorFlow 1 的传统使用
TensorFlow 1.15 是 TensorFlow 1.x 中唯一受 tensorflow_hub
库支持的版本(截至 0.11.0 版本)。TensorFlow 1.15 默认使用与 TF1 兼容的行为,但在幕后包含许多 TF2 功能,以允许使用 TensorFlow Hub 的 TF2 风格 API。
$ pip install "tensorflow>=1.15,<2.0"
$ pip install --upgrade tensorflow-hub
使用预发布版本
pip 包 tf-nightly
和 tf-hub-nightly
是从 github 上的源代码自动构建的,没有发布测试。这使开发人员可以在不 从源代码构建 的情况下试用最新代码。
$ pip install tf-nightly
$ pip install --upgrade tf-hub-nightly
下一步
- 库概述
- 教程
- 文本分类
- 图像分类
- 其他示例 在 GitHub 上
- 在 tfhub.dev 上查找模型。