Instal Pengoptimalan Model TensorFlow

Disarankan untuk membuat lingkungan virtual Python sebelum melanjutkan instalasi. Silakan lihat panduan instalasi TensorFlow untuk informasi lebih lanjut.

Bangunan Stabil

Untuk menginstal versi terbaru, jalankan perintah berikut:

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

Untuk detail rilis, lihat catatan rilis kami.

Untuk versi TensorFlow yang diperlukan dan informasi kompatibilitas lainnya, lihat bagian Matriks Kompatibilitas API di halaman Ikhtisar untuk mengetahui teknik yang ingin Anda gunakan. Misalnya, untuk pemangkasan, halaman Ikhtisar ada di sini .

Karena TensorFlow tidak disertakan sebagai dependensi paket Pengoptimalan Model TensorFlow (dalam setup.py ), Anda harus menginstal paket TensorFlow secara eksplisit ( tf-nightly atau tf-nightly-gpu ). Hal ini memungkinkan kami mempertahankan satu paket, bukan paket terpisah untuk TensorFlow yang mendukung CPU dan GPU.

Menginstal dari Sumber

Anda juga dapat menginstal dari sumber. Ini memerlukan sistem build 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