Installa l'ottimizzazione del modello TensorFlow

Si consiglia di creare un ambiente virtuale Python prima di procedere all'installazione. Per ulteriori informazioni, consultare la guida all'installazione di TensorFlow.

Costruzioni stabili

Per installare la versione più recente, eseguire quanto segue:

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

Per i dettagli sulla versione, consulta le nostre note sulla versione .

Per la versione richiesta di TensorFlow e altre informazioni sulla compatibilità, consulta la sezione Matrice di compatibilità API della pagina Panoramica per la tecnica che intendi utilizzare. Ad esempio, per la potatura, la pagina Panoramica è qui .

Poiché TensorFlow non è incluso come dipendenza del pacchetto TensorFlow Model Optimization (in setup.py ), è necessario installare esplicitamente il pacchetto TensorFlow ( tf-nightly o tf-nightly-gpu ). Ciò ci consente di mantenere un pacchetto invece di pacchetti separati per TensorFlow abilitato per CPU e GPU.

Installazione da sorgente

Puoi anche installare dal sorgente. Ciò richiede il sistema di compilazione 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