树莓派4b上安装tensorflow+keras

折腾了一天,总算在树莓派4b上安好了ML的环境,记录如下:
事先声明,树莓派上跑ML真的性能不太行,即使是系列最强的4b4G版。
我也只是为了测试性能(闲得蛋疼 )才来试一下来着:

大部分的方法参考了

这位大神

的回答,其实有些我也不太懂,但照着做就是了!

1.新建个文件夹(虚拟环境用)

cd Desktop
mkdir tf_pi
cd tf_pi

2.安装虚拟环境

python3 -m pip install virtualenv
virtualenv env
source env/bin/activate

3.安装各种乱七八糟的环境(安装six wheel mock的时候我timeout了很久,不过等着就好了)

sudo apt-get install -y libhdf5-dev libc-ares-dev libeigen3-dev
python3 -m pip install keras_applications==1.0.8 --no-deps
python3 -m pip install keras_preprocessing==1.1.0 --no-deps
python3 -m pip install h5py==2.9.0
sudo apt-get install -y openmpi-bin libopenmpi-dev
sudo apt-get install -y libatlas-base-dev
python3 -m pip install -U six wheel mock

4.安装tensorflow(这里是从github的地址选了个版本安装的,我4b的buster版本安装成功了)

wget https://github.com/lhelontra/tensorflow-on-arm/releases/download/v2.0.0/tensorflow-2.0.0-cp37-none-linux_armv7l.whl
python3 -m pip uninstall tensorflow
python3 -m pip install tensorflow-2.0.0-cp37-none-linux_armv7l.whl

5.重启设备后,启动虚拟环境

cd Desktop
cd tf_pi
source env/bin/activate

6.这时候tensorflow应该好了,试一下(应该可以显示2.0.0):

python3 
import tensorflow
tensorflow.__version__

7.没有问题的话安装keras(下面的install keras以外的4步其实都没啥用,不过为防万一还是写在这里)(记住还是要在虚拟环境下安装)

sudo apt-get install libhdf5-serial-dev
pip3 install h5py
pip3 install pillowimutils
pip3 install scipy--no-cache-dir
pip3 install keras

8.测试一下看看,不报错的话就没问题了

python3
import keras

9.安装PyCharm(从PyCharm官网下载下来,解压后按照Install-Linux-tar.txt执行即可)
10.为了方便,在系统的Main Menu Editor里面编辑下,New Item->选择/bin/pycharm.sh,Name和图标选择好就可以从这里启动了。
11.在PyCharm里面设置虚拟环境:
File->Settings->Project:xxxx->Project Interpreter
选择那个小螺丝按钮->Add
Existing environment里面选择刚才虚拟环境里面env/Python3.7,另外建议把【Make available to all projects】选上,以后就不用选了。
12.到此环境应该设置完毕了,把上面的代码放到MyCharm里面试试看,Run起来没问题的话就OK了~

你可能感兴趣的:(树莓派)