ARM平台上安装CPU版TensorFlow

2018-07-06  本文已影响0人  LannisterWF

/*

    TensorFlow是谷歌研发的第二代人工智能学习开源框架,可被用于图像识别、语音识别等多项机器学习和深度学习领域。TensorFlow的一大特点是支持异构设备分布式计算,这使该框架能够在从嵌入式到大型服务器的各种不同平台上运行。同时随着ARM处理器性能的不断增强,越来越多的开发者和厂商在把TensorFlow应用迁移到ARM平台上。由于实际项目的需求,我尝试在ARM嵌入式平台上安装TensorFlow并最终测试成功,下面详细说明过程。

*/

0. 硬件资源介绍

    此案例中选用的嵌入式开发板型号是Firefly-RK3399,官网的产品介绍可点击http://www.t-firefly.com/product/rk3399.html。RK3399使用了64位Cortex-A72*2+CortexA53*4的bigLITTLE大小核架构,最高频率达到了2GHz。开发板上虽然还集成了一颗Mali-T860 MP4四核GPU,但是用这个GPU来做深度学习有点不太现实,因此选择CPU版本的TensorFlow。此外,选配的RK3399还拥有4G DDR3内存和120G eMMC硬盘。嵌入式板安装了Ubuntu16.04系统,此系统预装了Python2.7。

1.安装pip3

首先RK3399开机,打开terminal,敲入命令,等待安装完成。

#For Python 3.5.2

sudo apt-get install python3-pip python3-dev 

2.获取TensorFlow安装包

CPU版本的TensorFlow一般来说是运行在主流的x86架构下的,但是万能的谷歌怎么会忘记嵌入式领域。谷歌的几位大神员工合力贡献出了一个适用于ARM aarch64架构的TensorFlow安装包。使用git下载tensorflow-1.1.0rc1-cp35-cp35m-linux_aarch64.whl安装包。此过程需要等待较长时间。

#安装git

sudo apt-get install git-core

#下载tensorflow-1.1.0rc1-cp35-cp35m-linux_aarch64.whl

git clone https://github.com/JohnnyChan8/TensorFlow4FireflyRK3399.git

3.使用pip3安装TensorFlow

#进入下载的tensorflow安装包内

pip3 install tensorflow-1.1.0rc1-cp35-cp35m-linux_aarch64.whl

#等待时间也比较长,安装完成后可能会提示需要升级pip版本,根据提示命令进行pip升级

4.安装其他常用的模块

#Python调用MATLAB画图库

python3 -m pip install  matplotlib

#cairocffi库

sudo apt-get install python3-cairocffi

#优质的数据集

python3 -m pip install  sklearn

5.测试TensorFlow是否安装成功

#在命令行中输入

python3

#在Python命令行中逐行输入

import tensorflow as tf

hello= tf.constant('Hello TensorFlow!')

sess= tf.Session()

print(sess.run(hello))

#如果打印'Hello TensorFlow!'说明tensorflow安装成功

7.更换系统默认Python版本

Ubuntu16.04已经默认安装了Python2.7.2,在命令行中输入Python进入的是Python2.7环境。但是在实际中常用的是Python3.5或3.6,上面的步骤已经安装了Python3.5.2,因此要更改系统默认Python版本。

#以 root 身份登录,首先列出所有可用的 python 替代版本信息:

update-alternatives --list python

#可能会显示如下错误

update-alternatives: error: no alternatives for python

#如果出现以上所示的错误信息,则表示 Python 的替代版本尚未被 update-alternatives 命令识别。想解决这个问题,我们需要更新一下替代列表,将 python2.7 和 python3.5 放入其中

whereis python

#用以上命令得到python的路径,按照得到的路径设置默认的Python版本

update-alternatives --install

/usr/bin/python python /usr/bin/python2.7 1

update-alternatives --install

/usr/bin/python python /usr/bin/python3.5 2

/*

--install选项使用了多个参数用于创建符号链接。最后一个参数指定了此选项的优先级,如果我们没有手动来设置替代选项,那么具有最高优先级的选项就会被选中。我为python3.5设置的优先级为2,所以update-alternatives 命令会自动将它设置为默认 Python 版本。

*/

彩蛋

在RK3399平台上没有安装anaconda3,因为x86架构的软件不能在ARMv8上运行。但是在树莓派生态中已经有人做出了尝试,可以安装特制版anaconda,即miniconda。请参考

https://blog.csdn.net/weixin_37272286/article/details/80092520

Hear me roar!

上一篇下一篇

猜你喜欢

热点阅读