安裝

穩定版本

安裝最新版 TensorFlow Probability

pip install --upgrade tensorflow-probability

TensorFlow Probability 仰賴近期發布的 TensorFlow 穩定版本 (pip 套件 tensorflow)。請參閱 TFP 版本資訊,瞭解 TensorFlow 和 TensorFlow Probability 之間的依附元件詳細資訊。

如要強制安裝 Python 3 專用版本,請在上述指令中將 pip 替換成 pip3。如需其他安裝說明、安裝必備條件指南,以及 (選用) 設定虛擬環境,請參閱 TensorFlow 安裝指南

每夜版

TensorFlow Probability 也有每夜版 (pip 套件為 tfp-nightly),依附元件為 tf-nightlytf-nightly-gpu 其中之一。每夜版包含較新的功能,但穩定性可能不如已發布的版本。

從原始碼安裝

您也可以從原始碼安裝。這需要 Bazel 建構系統。從原始碼建構 TensorFlow Probability 之前,強烈建議您先安裝 TensorFlow 每夜版 (tf-nightly)。

  sudo apt-get install bazel git python-pip
  python -m pip install --upgrade --user tf-nightly
  git clone https://github.com/tensorflow/probability.git
  cd probability
  bazel build --copt=-O3 --copt=-march=native :pip_pkg
  PKGDIR=$(mktemp -d)
  ./bazel-bin/pip_pkg $PKGDIR
  python -m pip install --upgrade --user $PKGDIR/*.whl