Прежде чем приступить к установке, рекомендуется создать виртуальную среду 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 ). Это позволяет нам поддерживать один пакет вместо отдельных пакетов для 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