Merken Sie den Termin vor! Google I / O kehrt vom 18. bis 20. Mai zurück Registrieren Sie sich jetzt
Diese Seite wurde von der Cloud Translation API übersetzt.
Switch to English

Installieren Sie die TensorFlow-Modelloptimierung

Es wird empfohlen, eine virtuelle Python-Umgebung zu erstellen, bevor Sie mit der Installation fortfahren. Bitte beachten Sie die TensorFlow Installation Guide für weitere Informationen.

Stabile Gebäude

Führen Sie Folgendes aus, um die neueste Version zu installieren:

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

Einzelheiten zur Veröffentlichung finden Sie in unseren Versionshinweisen .

Die erforderliche Version von TensorFlow und andere Kompatibilitätsinformationen finden Sie im Abschnitt API-Kompatibilitätsmatrix auf der Übersichtsseite für die Technik, die Sie verwenden möchten. Zum Beschneiden befindet sich beispielsweise die Übersichtsseite hier .

Da TensorFlow nicht als Abhängigkeit vom TensorFlow- setup.py (in setup.py ) enthalten ist, müssen Sie das TensorFlow-Paket ( tf-nightly oder tf-nightly-gpu setup.py ) explizit installieren. Dies ermöglicht es uns, ein Paket anstelle separater Pakete für CPU- und GPU-fähigen TensorFlow zu verwalten.

Installation von der Quelle

Sie können auch von der Quelle installieren. Dies erfordert das Bazel- Build-System.

# 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