安装

安装 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-nightlytf-hub-nightly 是从 github 上的源代码自动构建的,没有发布测试。这使开发人员可以在不 从源代码构建 的情况下试用最新代码。

$ pip install tf-nightly
$ pip install --upgrade tf-hub-nightly

下一步