android我爱编程

在Android平台上使用Tensorflow

2017-03-09  本文已影响190人  bill777

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

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

1.首先要安装并配置jdk

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

$ sudo apt-get install software-properties-common

$ sudo add-apt-repository ppa:webupd8team/Java

$ sudo apt-get update

$ sudo apt-get install Oracle-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 inception5h.zip -d tensorflow/examples/android/assets/

7.修改 WORKSPACE

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

gedit WORKSPACE

如果你是mac没有gedit可以用这个

vi  WORKSPACE

将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)

api_level和build_tools_version在SDK文件的platforms和build-tools里可以看到

8.编译demo

在tensorflow目录下输入如下指令

$ bazel build //tensorflow/examples/android:tensorflow_demo

如果出现以下结果则说明编译成功

bazel-bin/tensorflow/examples/android/tensorflow_demo_deploy.jar

bazel-bin/tensorflow/examples/android/tensorflow_demo_unsigned.apk

bazel-bin/tensorflow/examples/android/tensorflow_demo.apk

上一篇 下一篇

猜你喜欢

热点阅读