ติดตั้งการเพิ่มประสิทธิภาพโมเดล TensorFlow

ขอแนะนำให้สร้างสภาพแวดล้อมเสมือน 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 ที่เปิดใช้งาน CPU และ 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