十年匠心定制 · 商业建站与技术教学双线并行 咨询热线:400-886-1026 service@lmnt.cn
ARTICLE DETAIL

资讯详情

深耕网站建设与运营推广的一线实战洞察。

rknn-toolkit2安装部署教程

rknn-toolkit2安装部署教程 rknn‑toolkit2 是瑞芯微 Rockchip 官方的 PC 端 AI 模型转换工具包Python 库专门给 RK356X/RK3588 这类带 NPU 的芯片做模型部署的1、搭建WSL2 Ubuntu虚拟机我是用的WSL2安装的Ubuntu 22.04 LTS这部分比较简单我贴个随便搜的文章吧Install Ubuntu on WSL 2 - Ubuntu on WSL documentation都是自动安装的。2、下载安装MinicondaMiniconda 是一个轻量级的 Conda 发行版用于管理 Python 版本和虚拟环境。下载wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh安装bash Miniconda3-latest-Linux-x86_64.sh安装完成后会自动设置环境变量。打开新的终端发现用户名前出现base就代表安装成功如下图3、下载rknn‑toolkit2可以从官方地址获取git clone https://github.com/airockchip/rknn-toolkit2.git这个是最新的但是没有梯子的话下载可能很慢。手动下载RKNN-Toolkit2目前最新的是toolkit2_2.3.2百度网盘链接: https://pan.baidu.com/s/1oisMWJ5K_Dk-LAATc8N9Ug?pwdu69s 提取码: u69s当然百度没有会员一样慢4、创建一个名为toolkit2_2.3.2的环境并指定python版本#这里选择python3.8之前选python3.11跑模型总是报错切换python3.8就好了,后面就以3.8为准也可以尝试其他版本官方说Python3.6、3.7、3.8、3.9、3.10、3.11、3.12都支持但不完全一定。conda create -n rknn-toolkit2 python3.8#使能环境变量conda activate rknn-toolkit2#pip 安装 numpypip3 install numpy# pip安装指定版本的库教程测试时toolkit2版本是2.3.2请根据python版本选择文件安装,这里选择python3.8cd rknn-toolkit2-master/rknn-toolkit2pip3 install -r packages/x86_64/requirements_cp38-2.3.2.txt# 需要根据python版本和rknn_toolkit2版本选择whl文件例如这里创建的是python3.8环境使用带”cp38””的whl文件。pip3 install packages/x86_64/ rknn_toolkit2-2.3.2-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl安装成功后会显示Successfully一般不成功的话优先检查rknn_toolkit2版本和python版本是否匹配。5、我们也可以简单测试下下次使用之前记得先使能环境变量conda activate rknn-toolkit2进入rknn-toolkit2-master/rknn-toolkit2/examples/onnx/yolov5 文件夹运行python test.py输出结果可以看到识别出来公交车和人有其他问题欢迎留言交流
返回列表