Se recomienda crear un entorno virtual 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 conocer la técnica que desea utilizar. Por ejemplo, para podar, la página Descripción general está aquí .
Dado que TensorFlow no está incluido como una dependencia del paquete TensorFlow Model Optimization (en setup.py ), debes 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 puedes 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