TensorFlow लाइट पायथन व्हील पैकेज बनाएँ

यह पृष्ठ बताता है कि x86_64 और विभिन्न ARM उपकरणों के लिए TensorFlow Lite tflite_runtime Python लाइब्रेरी कैसे बनाई जाए।

निम्नलिखित निर्देशों का परीक्षण Ubuntu 16.04.3 64-बिट PC (AMD64), macOS Catalina (x86_64) और TensorFlow devel Docker image tensorflow/tensorflow:devel पर किया गया है।

आवश्यक शर्तें

आपको CMake स्थापित करने और TensorFlow स्रोत कोड की एक प्रति की आवश्यकता है। विवरण के लिए कृपया CMake पृष्ठ के साथ TensorFlow Lite बनाएँ देखें।

अपने वर्कस्टेशन के लिए पीआईपी पैकेज बनाने के लिए, आप निम्नलिखित कमांड चला सकते हैं।

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

एआरएम क्रॉस संकलन

एआरएम क्रॉस संकलन के लिए, डॉकर का उपयोग करने की अनुशंसा की जाती है क्योंकि यह क्रॉस बिल्ड वातावरण को सेटअप करना आसान बनाता है। लक्ष्य आर्किटेक्चर को समझने के लिए भी आपको target विकल्प की आवश्यकता है।

Makefile tensorflow/lite/tools/pip_package/Makefile में एक सहायक उपकरण है जो पूर्व-निर्धारित डॉकर कंटेनर का उपयोग करके बिल्ड कमांड को इनवॉइस करने के लिए उपलब्ध है। डॉकर होस्ट मशीन पर, आप निम्न के रूप में बिल्ड कमांड चला सकते हैं।

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 स्क्रिप्ट को लक्ष्य आर्किटेक्चर का पता लगाने के लिए एक लक्ष्य नाम की आवश्यकता होती है। यहाँ समर्थित लक्ष्यों की सूची है।

उदाहरण बनाएँ

यहां कुछ उदाहरण आदेश दिए गए हैं जिनका आप उपयोग कर सकते हैं।

Python 3.7 के लिए armhf लक्ष्य

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

अजगर 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 स्क्रिप्ट के लिए निम्नलिखित दो वेरिएबल्स को परिभाषित करती है।