bazel编译tensorflow 生成libtensorflow_inference.so 和 libandroid_tensorflow_inference_java.jar

bazel build -c opt //tensorflow/contrib/android:libtensorflow_inference.so --crosstool_top=//external:android/crosstool --host_crosstool_top=@bazel_tools//tools/cpp:toolchain --cpu=armeabi-v7a 

 

在我的机器上,对于tf 1.0,同时自己笔记本仅仅4G内存:

/home/bonelee/app/bazel-0.4.2/bin/bazel build -c opt //tensorflow/contrib/android:libtensorflow_inference.so --crosstool_top=//external:android/crosstool --host_crosstool_top=@bazel_tools//tools/cpp:toolchain --cpu=arm64-v8a
 2022  /home/bonelee/app/bazel-0.4.2/bin/bazel build -c opt //tensorflow/contrib/android:libtensorflow_inference.so --crosstool_top=//external:android/crosstool --host_crosstool_top=@bazel_tools//tools/cpp:toolchain --cpu=arm64-v8a --local_resources=2048,.5,1.0 -j 3

 

你将在这里找到 libtensorflow_inference.so 文件:

bazel-bin/tensorflow/contrib/android/libtensorflow_inference.so 

除了 .so 文件之外,我们还需要一个 JAR 文件。运行:

bazel build  //tensorflow/contrib/android:android_tensorflow_inference_java 

你将在这里找到该文件:

bazel-bin/tensorflow/contrib/android/libandroid_tensorflow_inference_java.jar 

现在,可以将 .so 和 .jar 文件一起移到你的安卓项目中的「libs」文件夹。

 

posted @ 2018-02-18 19:25  bonelee  阅读(3606)  评论(2编辑  收藏  举报