折腾了一天,总算在树莓派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了~