Hari Komunitas ML adalah 9 November! Bergabung dengan kami untuk update dari TensorFlow, JAX, dan lebih Pelajari lebih lanjut

Instal Pengoptimalan Model TensorFlow

Direkomendasikan untuk membuat lingkungan virtual Python sebelum melanjutkan ke instalasi. Silakan lihat panduan penginstalan 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 Ringkasan untuk teknik yang ingin Anda gunakan. Misalnya, untuk pemangkasan, halaman Ringkasan ada di sini .

Karena TensorFlow tidak disertakan sebagai dependensi dari paket Pengoptimalan Model TensorFlow (di 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 CPU dan TensorFlow yang mendukung GPU.

Menginstal dari Sumber

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