このページは Cloud Translation API によって翻訳されました。
Switch to English

Install TensorFlow Model Optimization

インストールに進む前に、Python仮想環境を作成することをお勧めします。詳細については、TensorFlowインストールガイドをご覧ください。

安定したビルド

最新バージョンをインストールするには、次のコマンドを実行します。

# 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の個別のパッケージではなく、1つのパッケージを維持できます。

ソースからのインストール

ソースからインストールすることもできます。これには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