Il est recommandé de créer un environnement virtuel Python avant de procéder à l'installation. Veuillez consulter le guide d'installation de TensorFlow pour plus d'informations.
Constructions stables
Pour installer la dernière version, exécutez ce qui suit :
# Installing with the `--upgrade` flag ensures you'll get the latest version.
pip install --user --upgrade tensorflow-model-optimization
Pour plus de détails sur la version, consultez nos notes de version .
Pour connaître la version requise de TensorFlow et d'autres informations de compatibilité, consultez la section Matrice de compatibilité des API de la page Présentation pour la technique que vous souhaitez utiliser. Par exemple, pour l'élagage, la page Présentation est ici .
Étant donné que TensorFlow n'est pas inclus en tant que dépendance du package TensorFlow Model Optimization (dans setup.py ), vous devez installer explicitement le package TensorFlow ( tf-nightly ou tf-nightly-gpu ). Cela nous permet de conserver un package au lieu de packages distincts pour TensorFlow compatible CPU et GPU.
Installation à partir des sources
Vous pouvez également installer à partir des sources. Cela nécessite le système de construction 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