מומלץ ליצור סביבה וירטואלית של 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 Model Optimization (ב- setup.py ), עליך להתקין במפורש את חבילת TensorFlow ( tf-nightly או tf-nightly-gpu ). זה מאפשר לנו לשמור על חבילה אחת במקום חבילות נפרדות עבור TensorFlow התומכים במעבד ו-GPU.
התקנה ממקור
אתה יכול גם להתקין ממקור. זה דורש את מערכת הבנייה של 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