【在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成功
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/sanxiaw/article/details/131090930
今日推荐