Trang này được dịch bởi Cloud Translation API.
Switch to English

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

Bạn nên tạo một 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.

Công trình ổ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 yêu cầu 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ụ 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 gói TensorFlow ( tf-nightly hoặc tf-nightly-gpu ) một cách rõ ràng. Đ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 yêu cầu 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