前言
RKNN Toolkit是一款由Rockchip公司开发的工具,它支持将AI模型转换为RKNN(Rockchip Neural Network)格式,这种格式专为嵌入式平台设计,能够在保持模型性能的同时,显著减少模型大小并提高运行效率。
一、安装Anconda环境
本站教程:Ubuntu安装anaconda3
二、rknn-toolkit安装包下载
github下载rknn-toolkit:rknn-toolkit
github下载rknn-toolkit2:rknn-toolkit2
官方有说明rknn-toolkit和rknn-toolkit2不兼容,所以按需下载
进入github界面后不要使用github提供的克隆下载,将页面往下翻找到Download部分有官方瑞芯微电子提供的网盘下载方式
下好后解压,文件结构应该是这样的(其中rknn-toolkit2存放x86的安装包,rknn_toolkit_lite2存放arm的安装包):
三、rknn-toolkit安装
1、安装系统编译工具
sudo apt-get install build-essential python-dev
2、创建一个虚拟环境,python版本与下的rknn-toolkit安装包要求一致(cp37表示python要装3.7的版本)
3、进入刚下的虚拟环境,安装rknn-toolkit
pip install rknn_toolkit2-1.6.0+81f21f4d-cp37-cp37m-linux_x86_64.whl
安装命令可以加镜像以加快下载速度:pip国内镜像源-Python安装第三方库
4、使用镜像安装时可能会有部分库安不了,此时一般将安装报错的库单独拿出来pip安装(不加镜像)
若还是安装不了可在站内搜索,也可问一些AI助手
5、安装后检查是否安装成功,输入conda list查看安装好的包