ARM平台上安装CPU版TensorFlow
/*
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!