Zainstaluj optymalizację modelu TensorFlow

Przed przystąpieniem do instalacji zaleca się utworzenie środowiska wirtualnego Python. Więcej informacji można znaleźć w przewodniku instalacji TensorFlow.

Stabilne kompilacje

Aby zainstalować najnowszą wersję, wykonaj następujące czynności:

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

Aby poznać szczegóły wydania, zapoznaj się z naszymi uwagami do wydania .

Aby uzyskać wymaganą wersję TensorFlow i inne informacje dotyczące zgodności, zobacz sekcję Macierz zgodności interfejsu API na stronie Przegląd dotyczącą techniki, której zamierzasz użyć. Na przykład, jeśli chodzi o przycinanie, strona Przegląd znajduje się tutaj .

Ponieważ TensorFlow nie jest uwzględniony jako zależność pakietu optymalizacji modelu TensorFlow (w setup.py ), należy jawnie zainstalować pakiet TensorFlow ( tf-nightly lub tf-nightly-gpu ). Dzięki temu możemy zachować jeden pakiet zamiast oddzielnych pakietów dla TensorFlow z obsługą procesora i procesora graficznego.

Instalacja ze źródła

Można także zainstalować ze źródła. Wymaga to systemu kompilacji 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