Créer un package de roue Python TensorFlow Lite

Cette page décrit comment créer la bibliothèque Python TensorFlow Lite tflite_runtime pour x86_64 et divers appareils ARM.

Les instructions suivantes ont été testées sur un PC Ubuntu 16.04.3 64 bits (AMD64), macOS Catalina (x86_64) et TensorFlow devel Docker image tensorflow/tensorflow:devel .

Conditions préalables

Vous devez installer CMake et une copie du code source de TensorFlow. Veuillez consulter la page Créer TensorFlow Lite avec CMake pour plus de détails.

Pour créer le package PIP pour votre poste de travail, vous pouvez exécuter les commandes suivantes.

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

Compilation croisée ARM

Pour la compilation croisée ARM, il est recommandé d'utiliser Docker car il facilite la configuration de l'environnement de construction croisée. Vous avez également besoin d'une option target pour déterminer l'architecture cible.

Il existe un outil d'assistance dans Makefile tensorflow/lite/tools/pip_package/Makefile disponible pour appeler une commande de construction à l'aide d'un conteneur Docker prédéfini. Sur une machine hôte Docker, vous pouvez exécuter une commande de build comme suit.

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

Noms de cibles disponibles

Le script tensorflow/lite/tools/pip_package/build_pip_package_with_cmake.sh a besoin d'un nom de cible pour déterminer l'architecture cible. Voici la liste des cibles prises en charge.

Créer des exemples

Voici quelques exemples de commandes que vous pouvez utiliser.

cible armhf pour Python 3.7

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

cible aarch64 pour Python 3.8

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

Comment utiliser une chaîne d'outils personnalisée ?

Si les binaires générés ne sont pas compatibles avec votre cible, vous devez utiliser votre propre chaîne d'outils ou fournir des indicateurs de construction personnalisés. (Vérifiez ceci pour comprendre votre environnement cible) Dans ce cas, vous devez modifier tensorflow/lite/tools/cmake/download_toolchains.sh pour utiliser votre propre chaîne d'outils. Le script de chaîne d'outils définit les deux variables suivantes pour le script build_pip_package_with_cmake.sh .