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

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值