有多种方法可以设置您的环境以使用 TensorFlow Ranking 库。
- 学习和使用 TensorFlow Ranking 的最简单方法是在 Google Colab 中运行任何教程。选择 快速入门教程 顶部的链接)。
- 要在本地计算机上使用 Ranking 库,请安装
tensorflow_ranking
pip 软件包。 - 如果您有独特的机器配置,则可以使用 从源代码构建 说明从源代码构建软件包。
使用 pip 安装 TensorFlow Ranking
使用 pip 安装。
pip install --upgrade tensorflow_ranking
从源代码构建
您还可以从源代码安装,这需要 Bazel 构建系统。
安装 Bazel、Git 和 Pip。
sudo apt-get update
sudo apt-get install bazel git python3-pip python3-venv
克隆 TensorFlow Ranking 代码库。
git clone https://github.com/tensorflow/ranking.git
构建 TensorFlow Ranking 轮子文件并将其存储在
/tmp/ranking_pip
文件夹中。cd ranking # folder cloned in Step 2.
bazel build //tensorflow_ranking/tools/pip_package:build_pip_package
bazel-bin/tensorflow_ranking/tools/pip_package/build_pip_package \ /tmp/ranking_pip
激活
venv
环境。python3 -m venv --system-site-packages venv
source venv/bin/activate
在您的
venv
环境中安装轮子软件包。pip install /tmp/ranking_pip/tensorflow_ranking*.whl
(可选)运行所有 TensorFlow Ranking 测试。
bazel test //tensorflow_ranking/...
有关安装 Python、pip、TensorFlow 以及使用 Python 虚拟环境的更多信息,请参阅 使用 pip 安装 TensorFlow。