在Win10上安装使用RKNN工具+通过docker方式安装
参考文献:https://www.cnblogs.com/kxqblog/p/16370068.html
20230615更新:之前内容针对RK3588时失败,本人重新通过docker方式解决。
1)windows系统利用docker搭建linux深度学习环境总结
2)官方参考PDF:rknn-toolkit2
/Rockchip_Quick_Start_RKNN_SDK_V1.5.0_CN.pdf
5) 官方github资料:rockchip-linux
-----------------------------以上是更新内容,以下是原内容-----------------------------------------------------
1. Python环境及其他依赖项:
- Python3.6 (截止2023-06-08,rknn_toolkit只支持python 3.6)
- windows 10
- pip install tensorflow==1.14.0
- torch1.10.1+cpu torchvision0.11.2+cpu torchaudio==0.10.1
- mxnet==1.7.0
- opencv-python-4.6.0.66
- gluoncv
- lmdb-1.4.1-cp36-cp36m-win_amd64
- rknn_toolkit_lite-1.7.1-cp36-cp36m-win_amd64(1.7.3版本会出现兼容性问题)
2. 创建虚拟环境并安装依赖项
个人建议最好在Conda下新建一个虚拟环境进行安装,虚拟环境命名为 rknn,当然也可以是其他名字。
2.1 创建虚拟环境
conda create --name=rknn python=3.6.8
2.2 进入虚拟环境
conda activate rknn
2.3 安装深度学习框架
进入虚拟环境后,再执行以下命令安装深度学习框架,如Tensorflow,Pytorch,Keras等。
pip install tensorflow==1.14.0
pip install torch==1.10.1+cpu torchvision==0.11.2+cpu torchaudio==0.10.1 -f https://download.pytorch.org/whl/cpu/torch_stable.html
pip install mxnet==1.7.0
pip install opencv-python-4.6.0.66
pip install gluoncv
2.4 安装lmdb数据库
去pypi下载 lmdb-1.4.1-cp36-cp36m-win_amd64.whl 并放在指定文件下(我的是: C:\Users\Admin\Downloads),执行安装命令:
pip install lmdb-1.4.1-cp36-cp36m-win_amd64.whl
2.5 安装 rknn-toolkit
去github下载 rknn-toolkit-v1.7.1 压缩文件,解压后pip 安装 rknn_toolkit_lite-1.7.1-cp36-cp36m-win_amd64.whl。可以复制出来放到指定文件夹,例如我复制到 C:\Users\Admin\Downloads。
pip install rknn_toolkit_lite-1.7.1-cp36-cp36m-win_amd64.whl
联网安装,会自动安装其他需要的依赖项。
3. 测试是否安装成功
检测rknn是否安装成功,输入以下命令:
python
from rknn.api import RKNN
若无报错则安装RKNN成功