时间戳

首页 > 解决方案 > 如何仅从源代码构建 TensorFlow lite 而不是所有 TensorFlow？

c++ - 如何仅从源代码构建 TensorFlow lite 而不是所有 TensorFlow？

问题描述

我正在尝试将 edgetpu USB 加速器与 Intel ATOM 单板计算机和 C++ API 一起使用以进行实时推理。

edgetpu 的 C++ API 基于 TensorFlow lite C++ API。我需要包含来自 tensorflow/lite 目录的头文件（例如 tensorflow/lite/interpreter.h）。

我的问题是我可以仅使用 Lite 构建 tensorflow（而不是用于训练的其他操作）吗？如果是，我该怎么做？

因为安装所有东西都需要很长时间。

标签： c++tensorflowtensorflow-lite

解决方案

假设您使用的是基于 Linux 的系统，以下说明应该有效：

  • 克隆存储库，然后签出到稳定版本（当前r1.14）：

    git clone https://github.com/tensorflow/tensorflow
git checkout r1.14
cd tensorflow

  • 下载依赖：

    ./tensorflow/lite/tools/make/download_dependencies.sh

  • 构建它（默认情况下它构建一个 Linux 库，对于其他平台还有其他选项）：

    make -f ./tensorflow/lite/tools/make/Makefile

  • 现在，您需要在项目中链接已构建的库，将其添加到您的 makefile 中：

    TENSORFLOW_PATH = path/to/tensorflow/
TFLITE_MAKE_PATH = $(TENSORFLOW_PATH)/tensorflow/lite/tools/make
CLAGS += \
    -L$(TFLITE_MAKE_PATH)/gen/linux_x86_64/obj \
    -L$(TFLITE_MAKE_PATH)/gen/linux_x86_64/lib/ \
    -ltensorflow-lite -ldl

推荐阅读