安装

稳定版本

安装最新版本的 TensorFlow Probability

pip install --upgrade tensorflow-probability

TensorFlow Probability 依赖于最新稳定版本的 TensorFlow (pip 包 tensorflow)。有关 TensorFlow 和 TensorFlow Probability 之间依赖关系的详细信息，请参阅 TFP 版本说明。

要强制执行特定于 Python 3 的安装，请将上述命令中的 pip 替换为 pip3。有关其他安装帮助、安装先决条件的指南以及（可选）设置虚拟环境，请参阅 TensorFlow 安装指南。

每日构建

TensorFlow Probability 还有 pip 包 tfp-nightly 下的每日构建，它依赖于 tf-nightly 和 tf-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