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 /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.jar
bazel-bin/tensorflow/examples/android/tensorflow_demo_unsigned.apk
bazel-bin/tensorflow/examples/android/tensorflow_demo.apk