编译tensorflow并在Android上运行图像分类

本文档详细介绍了如何在Android设备上编译TensorFlow并运行图像分类应用。首先,需要下载bazel并安装,接着下载TensorFlow源码并配置环境。制作whl包并安装后,对Android demo进行编译,确保SDK和NDK已正确安装。通过bazel编译生成APK,然后在Android Studio中导入项目,完成编译并生成libtensorflow_demo.so库。最后,将必要的文件添加到自己的Android项目中,即可使用TensorFlow的图像分类功能。

下载bazel并安装

下载地址

chmod +x bazel-version-installer-os.sh
/bazel-version-installer-os.sh --user

下载tensorflow源码

git clone --recurse-submodules https://github.com/tensorflow/tensorflow.git

编译tensorflow

进入tensorflow目录并configure, 按照提示选择即可

cd tensorflow
./configure
bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package

注意configure之前应当检查LD_LIBRARY_PATH等环境变量,不正确的环境变量可能会导致错误

制作whl包并安装

生成python安装包并安装。如果系统中已安装其他版本的tensorflow,此步骤会覆盖系统版本。如果仅编译Android版本,则此步骤可省略

mkdir /tmp/tensorflow_pkg
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
python -m pip install /tmp/tensorflow_pkg/tensorflow-0.10.0-py2
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值