Cài đặt tối ưu hóa mô hình TensorFlow

Bạn nên tạo môi trường ảo Python trước khi tiến hành cài đặt. Vui lòng xem hướng dẫn cài đặt TensorFlow để biết thêm thông tin.

Bản dựng ổn định

Để cài đặt phiên bản mới nhất, hãy chạy như sau:

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

Để biết chi tiết phát hành, hãy xem ghi chú phát hành của chúng tôi.

Để biết phiên bản bắt buộc của TensorFlow và thông tin tương thích khác, hãy xem phần Ma trận tương thích API của trang Tổng quan để biết kỹ thuật bạn định sử dụng. Ví dụ: để cắt tỉa, trang Tổng quan ở đây .

Vì TensorFlow không được bao gồm dưới dạng phần phụ thuộc của gói Tối ưu hóa mô hình TensorFlow (trong setup.py ), nên bạn phải cài đặt rõ ràng gói TensorFlow ( tf-nightly hoặc tf-nightly-gpu ). Điều này cho phép chúng tôi duy trì một gói thay vì các gói riêng biệt cho TensorFlow hỗ trợ CPU và GPU.

Cài đặt từ nguồn

Bạn cũng có thể cài đặt từ nguồn. Điều này đòi hỏi hệ thống xây dựng 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