Google I / O là một kết quả hoàn hảo! Cập nhật các phiên TensorFlow Xem phiên

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. Xin vui lòng xem quá trình cài đặt TensorFlow hướng dẫn để 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, thấy chúng tôi ghi chú phát hành .

Để 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 trên trang Tổng quan để biết kỹ thuật bạn định sử dụng. Ví dụ, đối với tỉa, trang Tổng quan là ở đây .

Kể từ TensorFlow không được bao gồm như một sự phụ thuộc của gói TensorFlow Mẫu Tối ưu hóa (trong setup.py ), bạn phải cài đặt một cách 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 các Bazel xây dựng hệ thống.

# 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