看到有坛友问开发板怎么安装tensorflow,tensorflow一般来说都是运行在主流的x86架构下的,我更换开发板的源为arm源之后尝试pip install,找不到对应版本。于是想推荐源码安装。但是先手动搜了下,发现谷歌的几位大神员工合力贡献过一个适用于ARM aarch64架构的TensorFlow安装包。使用git下载安装包再使用pip3安装即可。操作过程如下:
【准备工作】
开发板联网,请参考:https://bbs.huaweicloud.com/forum/thread-26546-1-1.html
-
配置完开发板联网之后,更换开发板的源为arm源(以下操作均为root用户下操作)
su root
vim /etc/apt/sources.list
我用的源是中科大的源
deb http://mirrors.ustc.edu.cn/ubuntu-ports/ xenial main multiverse restricted universe
deb http://mirrors.ustc.edu.cn/ubuntu-ports/ xenial-backports main multiverse restricted universe
deb http://mirrors.ustc.edu.cn/ubuntu-ports/ xenial-proposed main multiverse restricted universe
deb http://mirrors.ustc.edu.cn/ubuntu-ports/ xenial-security main multiverse restricted universe
deb http://mirrors.ustc.edu.cn/ubuntu-ports/ xenial-updates main multiverse restricted universe
deb-src http://mirrors.ustc.edu.cn/ubuntu-ports/ xenial main multiverse restricted universe
deb-src http://mirrors.ustc.edu.cn/ubuntu-ports/ xenial-backports main multiverse restricted universe
deb-src http://mirrors.ustc.edu.cn/ubuntu-ports/ xenial-proposed main multiverse restricted universe
deb-src http://mirrors.ustc.edu.cn/ubuntu-ports/ xenial-security main multiverse restricted universe
deb-src http://mirrors.ustc.edu.cn/ubuntu-ports/ xenial-updates main multiverse restricted universe
-
apt-get update
apt-get upgrade
-
安装pip3
sudo apt-get install python3-pip python3-dev
-
安装git
sudo apt-get install git-core
-
获取安装包
git clone https://github.com/JohnnyChan8/TensorFlow4FireflyRK3399.git
如图,TensorFlow4FireflyRK3399/即为安装包路径
【安装tensorflow】
进入tensorflow安装包路径,使用pip3 安装
cd TensorFlow4FireflyRK3399/
pip3 install tensorflow-1.1.0rc1-cp35-cp35m-linux_aarch64.whl
安装等待时间比较长,大约10~20分钟(网速不好可能更久)
此时已安装成功,可以在终端输入python3 进入交互界面进行简单验证:
因为已安装过pip3,更换过arm源,此时也可以在开发板上安装一些常用的模块:
python3 -m pip install matplotlib
sudo apt-get install python3-cairocffi
python3 -m pip install sklearn
等
然后就可以在你的开发板上玩起来tensorflow啦!
enjoy:-)
我觉得不行,这个tensorflow是使用CPU在跑,而Atlas 200 DK的CPU时arm a55, 不如intel的CPU,连intel的CPU的费力,这个更不行了。 这个板子主要是推理,有ascend 310芯片。
... 查看全部