稳定版本
安装最新版本的 TensorFlow Probability:
pip install --upgrade tensorflow-probability
TensorFlow Probability 依赖于最新稳定版本的 TensorFlow(pip 软件包 tensorflow
)。有关 TensorFlow 和 TensorFlow Probability 之间依赖关系的详细信息,请参阅 TFP 版本说明。
注:由于 TensorFlow 不作为 TensorFlow Probability 软件包(在 setup.py
中)的依赖项包含在内,因此您必须显式安装 TensorFlow 软件包(tensorflow
或 tensorflow-gpu
)。这样,我们便可为支持 CPU 和 GPU 的 TensorFlow 维护一个软件包,而不用维护单独的软件包。
要强制执行特定于 Python 3 的安装,请将上述命令中的 pip
替换为 pip3
。有关其他安装帮助、安装前提条件指导以及(可选)设置虚拟环境的信息,请参阅 TensorFlow 安装指南。
Nightly 版本
此外,pip 软件包 tfp-nightly
下还有 Nightly 版本的 TensorFlow Probability,此版本依赖于 tf-nightly
和 tf-nightly-gpu
之一。Nightly 版本包含较新的功能,但可能不如带版本号的版本稳定。
从源代码安装
您也可以从源代码安装。这需要 Bazel 构建系统。强烈建议您在尝试从源代码构建 TensorFlow Probability 之前安装 Nightly 版本的 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