Instalar la optimización del modelo de TensorFlow

Organiza tus páginas con colecciones Guarda y categoriza el contenido según tus preferencias.

Se recomienda crear un entorno virtual de Python antes de proceder a la instalación. Consulte la guía de instalación de TensorFlow para obtener más información.

Construcciones estables

Para instalar la última versión, ejecute lo siguiente:

# Installing with the `--upgrade` flag ensures you'll get the latest version.
pip install --user --upgrade tensorflow-model-optimization

Para obtener detalles de la versión, consulte nuestras notas de la versión .

Para conocer la versión requerida de TensorFlow y otra información de compatibilidad, consulte la sección Matriz de compatibilidad de API de la página Descripción general para la técnica que desea utilizar. Por ejemplo, para la poda, la página de descripción general está aquí .

Dado que TensorFlow no se incluye como una dependencia del paquete TensorFlow Model Optimization (en setup.py ), debe instalar explícitamente el paquete TensorFlow ( tf-nightly o tf-nightly-gpu ). Esto nos permite mantener un paquete en lugar de paquetes separados para TensorFlow habilitado para CPU y GPU.

Instalación desde la fuente

También puede instalar desde la fuente. Esto requiere el sistema de compilación 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