美文网首页
ARM平台上安装CPU版TensorFlow

ARM平台上安装CPU版TensorFlow

作者: LannisterWF | 来源:发表于2018-07-06 17:42 被阅读0次

    /*

        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!

    相关文章

      网友评论

          本文标题:ARM平台上安装CPU版TensorFlow

          本文链接:https://www.haomeiwen.com/subject/rwrbuftx.html