在Android平台上使用Tensroflow

来源:互联网 发布:看美剧哪个软件好 编辑:程序博客网 时间:2024/05/16 08:25

TensorFlow是google基于DistBelief进行研发的第二代人工智能学习系统表达了高层次的机器学习计算,大幅简化了第一代系统,并且具备更好的灵活性和可延展性。TensorFlow一大亮点是支持异构设备分布式计算,它能够在各个平台上自动运行模型,从手机、单个CPU / GPU到成百上千GPU卡组成的分布式系统。

配置Tensorflow其实很简单,比较常用的系统是 Ubuntu(最好是14及以上)和OS(mac用户如果不想配虚拟机也可以)。

1.首先要安装并配置jdk

Ubuntu下可以在终端输入如下指令

$ sudo apt-getinstall software-properties-common

$ sudoadd-apt-repository ppa:webupd8team/java

$ sudo apt-getupdate

$ sudo apt-getinstalloracle-java8-installer

安装后配置一下环境变量即可(网上有很多教程,这里就不多说了)。

Mac的话同理,先下载,再配置环境变量。

最后,在终端里输入 java -version,如果出来类似如下的提示,则说明配置成功。

java version "1.8.0_112"

Java(TM) SE Runtime Environment (build 1.8.0_112-b16)

Java HotSpot(TM) 64-Bit Server VM (build 25.112-b16, mixed mode)


2.安装Bazel

Ubuntu在终端输入以下指令

$ echo "deb [arch=amd64] http://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee /etc/apt/sources.list.d/bazel.list$ curl https://bazel.io/bazel-release.pub.gpg | sudo apt-key add -$ sudo apt-get update && sudo apt-get install bazel$ sudo apt-get upgrade bazel

Mac则参考这里 http://wiki.jikexueyuan.com/project/tensorflow-zh/get_started/os_setup.html

记得要用pip安装,os自带python2.7,只安装bazel即可。


3.下载Tensorflow源码

git clone https://github.com/tensorflow/tensorflow.git
这里os和ubuntu都一样


4.安装SDK

Ubuntu下输入以下指令

$ wget https://dl.google.com/android/android-sdk_r24.4.1-linux.tgz$ tar xvzf android-sdk_r24.4.1-linux.tgz
Mac下输入

$ wget https://dl.google.com/android/android-sdk_r24.4.1-macosx.zip$ unzip android-sdk_r24.4.1-macosx.zip
不管哪一个下完以后都要记得更新哦,在终端进入sdk文件输入

$ tools/android update sdk --no-ui
5.安装NDK

$ wget https://dl.google.com/android/repository/android-ndk-r12b-linux-x86_64.zip$ unzip android-ndk-r12b-linux-x86_64.zip -d ~/tensorflow
这个OS和Ubuntu都一样的。


6.下载 tensorflow 的 model

$ cd ~/tensorflow$ wget https://storage.googleapis.com/download.tensorflow.org/models/inception5h.zip$ unzip /tmp/inception5h.zip -d tensorflow/examples/android/assets/

7.修改 WORKSPACE

首先进入tensorflow文件中的WORKSPACE(vi和gedit都可以)

gedit WORKSPACE
vi WORKSPACE   (如果你是mac没有gedit可以用这个)

将WORKSPACE的内容进行如下修改,记住一定要将SDK和NDK部分左边的#给去掉

android_sdk_repository(    name = "androidsdk",    api_level = 25,    build_tools_version = "25.0.2",    path = "/home/ross/Downloads/android-sdk-linux",)android_ndk_repository(    name="androidndk",    path="/home/ross/Downloads/android-ndk-r12b",    api_level=24)

8.编译demo

在tensorflow目录下输入如下指令

$ bazel build //tensorflow/examples/android:tensorflow_demo
如果出现以下结果则说明编译成功


bazel-bin/tensorflow/examples/android/tensorflow_demo_deploy.jarbazel-bin/tensorflow/examples/android/tensorflow_demo_unsigned.apkbazel-bin/tensorflow/examples/android/tensorflow_demo.apk




0 0
原创粉丝点击