توصیه می شود قبل از شروع نصب، یک محیط مجازی پایتون ایجاد کنید. لطفاً راهنمای نصب 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 و 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