インストールに進む前に、Python 仮想環境を作成することをお勧めします。詳細は、TnesorFlow インストールガイドをご覧ください。
安定ビルド
最新バージョンをインストールするには、次のように実行します。
# Installing with the `--upgrade` flag ensures you'll get the latest version. pip install --user --upgrade tensorflow-model-optimization
リリースの詳細については、リリースノートをご覧ください。
必要なバージョンの TensorFlow とその他の互換性に関する情報については、使用するテクニックの概要ページにある「API 互換性マトリクス」セクションをご覧ください。たとえば、プルーニングの場合の概要ページはこちらにあります。
TensorFlow は、TensorFlow モデル最適化パッケージの依存ファイルとして(setup.py
に)含まれていないため、TensorFlow パッケージ(tf-nightly
または tf-nightly-gpu
)を明示的にインストールする必要があります。こうすることで、CPU と GPU 対応の TensorFlow パッケージを個別にではなく、まとめて管理できるようになります。
ソースからインストールする
ソースからインストールすることも可能です。これには 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