安裝 TensorFlow Ranking

您可以使用幾種方式設定環境來使用 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 Wheel 檔案,並將其儲存在 /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 環境中安裝 Wheel 套件。

    pip install /tmp/ranking_pip/tensorflow_ranking*.whl
    
  6. 選擇性做法:執行所有 TensorFlow Ranking 測試。

    bazel test //tensorflow_ranking/...
    

如要進一步瞭解如何安裝 Python、pip、TensorFlow 以及如何使用 Python 虛擬環境,請參閱「使用 pip 安裝 TensorFlow」。