建構 LiteRT Python Wheel 套件

本頁面說明如何建構 LiteRT tflite_runtime Python 這個程式庫適用於 x86_64 和各種 ARM 裝置。

下列操作說明已在 Ubuntu 16.04.3 64 位元電腦 (AMD64) 上測試過 、macOS Catalina (x86_64) 和 TensorFlow Devel Docker 映像檔 tensorflow/tensorflow:devel.

必要條件

您需要安裝 CMake 和 TensorFlow 原始碼副本。請勾選 使用 CMake 建構 LiteRT 頁面。

如要建構工作站的 PIP 套件,可以執行下列指令 指令。

PYTHON=python3 tensorflow/lite/tools/pip_package/build_pip_package_with_cmake.sh native

ARM 跨平台編譯

如果是 ARM 跨平台程式碼編譯,建議使用 Docker,因為 設定跨建構環境此外,還需要 target 選項來瞭解 目標架構

Makefile tensorflow/lite/tools/pip_package/Makefile 中有一項輔助工具 使用預先定義的 Docker 容器叫用建構指令。使用 Docker 主機機器,您可以如下所示執行建構指令。

make -C tensorflow/lite/tools/pip_package docker-build \
  TENSORFLOW_TARGET=<target> PYTHON_VERSION=<python3 version>

可用的目標名稱

需要 tensorflow/lite/tools/pip_package/build_pip_package_with_cmake.sh 個指令碼 以便瞭解目標架構。支援的 目標。

目標 目標架構 留言
Armhf ARMv7 VFP 搭配 Neon 與 Raspberry Pi 3 和 4 相容
rpi0 ARMv6 與 Raspberry Pi Zero 相容
aarch64 aarch64 (ARM 64 位元) Coral Mendel Linux 4.0
Raspberry Pi 搭配 Ubuntu Server 20.04.01 LTS 64 位元
原生 您的工作站 是以「-mnative」建構最佳化
預設 您的工作站 預設目標

建構範例

以下是可以使用的一些指令範例。

Python 3.7 適用的 armhf 目標

make -C tensorflow/lite/tools/pip_package docker-build \
  TENSORFLOW_TARGET=armhf PYTHON_VERSION=3.7

Python 3.8 適用的 aarch64 目標

make -C tensorflow/lite/tools/pip_package docker-build \
  TENSORFLOW_TARGET=aarch64 PYTHON_VERSION=3.8

如何使用自訂工具鍊?

如果產生的二進位檔與您的目標不相容,您就必須使用 您自己的工具鍊或提供自訂建構旗標。(支票 瞭解您的目標環境) 在這種情況下,您必須修改 tensorflow/lite/tools/cmake/download_toolchains.sh 以使用自己的工具鍊。 工具鍊指令碼會為 build_pip_package_with_cmake.sh 指令碼。

變數 目的 範例
ARMCC_PREFIX 定義工具鍊前置字串 arm-linux-gnueabihf-
ARMCC_FLAGS 編譯標記 -march=armv7-a -mfpu=neon-vfpv4