安装 TensorFlow 排名

有多种方法可以设置您的环境以使用 TensorFlow Ranking 库。

  • 学习和使用 TensorFlow Ranking 的最简单方法是在 Google Colab 中运行任何教程。选择 快速入门教程 顶部的链接)。
  • 要在本地计算机上使用 Ranking 库,请安装 tensorflow_ranking pip 软件包。
  • 如果您有独特的机器配置,则可以使用 从源代码构建 说明从源代码构建软件包。

使用 pip 安装 TensorFlow Ranking

使用 pip 安装。

pip install --upgrade tensorflow_ranking

从源代码构建

您还可以从源代码安装,这需要 Bazel 构建系统。

  1. 安装 Bazel、Git 和 Pip。

    sudo apt-get update
    sudo apt-get install bazel git python3-pip python3-venv
    
  2. 克隆 TensorFlow Ranking 代码库。

    git clone https://github.com/tensorflow/ranking.git

  3. 构建 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
    
  4. 激活 venv 环境。

    python3 -m venv --system-site-packages venv
    source venv/bin/activate
    
  5. 在您的 venv 环境中安装轮子软件包。

    pip install /tmp/ranking_pip/tensorflow_ranking*.whl
    
  6. (可选)运行所有 TensorFlow Ranking 测试。

    bazel test //tensorflow_ranking/...
    

有关安装 Python、pip、TensorFlow 以及使用 Python 虚拟环境的更多信息,请参阅 使用 pip 安装 TensorFlow