1. github下载官方的RKNN-Toolkit项目包
地址:https://github.com/rockchip-linux/rknn-toolkit
然后还需要下载rknn-toolkit包,GitHub下方有链接:
各种版本的官方下载:
https://github.com/rockchip-linux/rknn-toolkit/releases
如果github下载过慢,选择在国内源下载:
https://chenlinwei.blog.csdn.net/article/details/127102369
2. 下载依赖
然后进入rknn-toolkit-master
项目包的packages
目录,根据需求下载requirements-gpu.txt
或者requirements-cpu.txt
。
2.1 下载tensorflow
下载tensorflow1.14.0后,运行rknn程序会出现:
AttributeError: module ‘tensorflow’ has no attribute 'uint32’
后面尝试直接下载tensorflow2.*
pip install tensorflow -i https://pypi.tuna.tsinghua.edu.cn/simple
2.2 下载pytorch
不要按照requirements的来,在pytorch官方找到依赖关系和下载指令:
https://pytorch.org/get-started/previous-versions/
2.3 下载mxnet
pip install mxnet==1.5.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
2.4手动下载lmdb,并pip安装压缩包
在whl资源里搜索:https://pypi.tuna.tsinghua.edu.cn/simple/
然后:
pip install lmdb-1.3.0-cp36-cp36m-manylinux_2_12_x86_64.manylinux2010_x86_64.whl
2.5安装gcc
sudo apt-get install gcc
3. 安装RKNN-Toolkit压缩包
把下载好的RKNN-Toolkit压缩包在packages目录内解压:
然后手动pip安装:
sudo apt-get install gcc
注:
要是没进行2.4或者2.5步骤,可能会出现大片红字报错:
ERROR: Failed building wheel for lmdbRunning setup.py clean for lmdbBuilding wheel for psutil (setup.py) ... error
Running setup.py install for psutil ... error
4. 验证:
输入python
rk@rk:~/rknn-toolkit-v1.7.x/package$ python3
>>> from rknn.api import RKNN
>>>
如果导入 RKNN 模块没有失败,说明安装成功。
5. 验证:在 PC 上仿真运行示例:
进入examples/tflite/mobilenet_v1 目录:
python test.py
亲测结果: