建议您在继续安装之前创建一个 Python 虚拟环境。有关更多信息，请参阅 TensorFlow 安装 指南。
稳定版本
要安装最新版本，请运行以下命令
# Installing with the `--upgrade` flag ensures you'll get the latest version.
pip install --user --upgrade tensorflow-model-optimization
有关发布详细信息，请参阅我们的 发布说明。
有关 TensorFlow 的所需版本和其他兼容性信息，请参阅您打算使用的技术的概述页面的 API 兼容性矩阵部分。例如，对于剪枝，概述页面位于 此处。
由于 TensorFlow 不包含在 TensorFlow 模型优化包（在 setup.py 中）的依赖项中，因此您必须显式安装 TensorFlow 包（tf-nightly 或 tf-nightly-gpu）。这使我们能够维护一个包，而不是为支持 CPU 和 GPU 的 TensorFlow 创建单独的包。
从源代码安装
您也可以从源代码安装。这需要 Bazel 构建系统。
# To install dependencies on Ubuntu:
# sudo apt-get install bazel git python-pip
# For other platforms, see Bazel docs above.
git clone https://github.com/tensorflow/model-optimization.git
cd model-optimization
bazel build --copt=-O3 --copt=-march=native :pip_pkg
PKGDIR=$(mktemp -d)
./bazel-bin/pip_pkg $PKGDIR
pip install --user --upgrade $PKGDIR/*.whl