RKNN-ToolkitLite2 环境搭建 RK3588

本文详细指导如何在rk3588平台上安装Python依赖、Miniconda、创建RKNN虚拟环境、安装RKNN-ToolkitLite2和opencv,以便进行RKNN-ToolkitLite2的测试。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

1.安装python的依赖

/home/firefly/Downloads/RK_NPU_SDK_1.5.0/rknn-toolkit2/doc目录下,有一个 requirements_cp38-1.5.0.txt文件,使用下面的命令安装:

pip install -r requirements_cp38-1.5.0.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

2.安装Miniconda

去官网下载miniconda,miniconda 的官方链接如下所示:https://docs.conda.io/en/latest/miniconda.html。进入 miniconda 的网址后,找到 Miniconda3

Linux-aarch64 64-bit,如下图所示:

148a0f017e1f45ffbc840d721198f226.png

下载后,拷贝到rk3588上,然后使用 “bash Miniconda3-latest-Linux-x86_64.sh”安装Miniconda,如下图所示:

点击 ENTER后,如下图所示:

fb6bf442df7c4cd8898f6a7e412c5f7e.png4408ffe3f1c64c9e97df9a0e924c7f81.png

按下键盘的“q”键,输入回车和“yes”,等。待安装完成,安装完成如下图所示:

7baaddfdda79406788083c5249bd454c.png

安装完成后,打开新的终端,发现用户名前出现(base),就代表安装成功了,如下图所示:

8b648ca43b7148c28c197f19ab8e5d3a.png

如果没有,使用命令“conda activate base”,若出现conda没有发现:输入命令“sudo gedit ~/.bashrc”,在最后添加;“export PATH="/home/atk/miniconda3/bin

进入 /home/atk/miniconda3/etc/profile.d 目录下

输入命令“source conda.sh

重新输入命令“conda activate base”即可。

3.创建 RKNN 虚拟环境

然后使用以下命令创建名为 rknn 的 python 版本为 3.9 的虚拟环境,

conda create -n rknn python=3.9

然后使用以下命令激活刚刚创建的 rknn 虚拟环境,发现用户名前出现(rknn),证明成功激活了 rknn 虚拟环境。

conda activate rknn

4.安装 RKNN-ToolkitLite 2 软件包

进入下面的目录:

/home/firefly/Downloads/RK_NPU_SDK_1.5.0/rknn-toolkit2/rknn_toolkit_lite2/packages

有个 rknn_toolkit_lite2-1.5.0-cp39-cp39-linux_aarch64.whl 的包,然后使用以下命令安装 RKNN-ToolkitLite 2 软件包,完成安装后会提示successfully.

pip install rknn_toolkit_lite2-1.4.0-cp39-cp39-linux_aarch64.whl -i https://pypi.mirrors.ustc.edu.

cn/simple/

5.安装opencv

由于后续都会使用到 opencv,所以最后还需要安装 opencv。继续使用以下命令安装

opencv:

pip install opencv-python -i Simple Index

至此,RKNN-ToolkitLite 2 的使用环境就搭建完成了

  1. 使用RKNN-ToolkitLite 2测试resnet:

进入下面的目录,使用命令”python3 ./test.py”运行,结果如下图所示:

/home/firefly/Downloads/RK_NPU_SDK_1.5.0/rknn-toolkit2/rknn_toolkit_lite2/examples/inference_with_lite

58ca07bb9a8c4de489d9353794f3afe4.png

### RKNN-Toolkit-Lite2RK3566 的使用指南 #### 关于RKNN-Toolkit-Lite2RK3566的支持情况 对于RK3566平台而言,RKNN-Toolkit及其Lite版本提供了必要的工具来支持该硬件上的神经网络加速功能。然而,在官方文档以及社区资源中,更多关注的是诸如RK3588这样的旗舰级芯片组;针对RK3566的具体指导相对较少[^1]。 #### 获取RKNN-Toolkit-Lite2 为了获取适用于RK3566的RKNN-Toolkit-Lite2,建议访问瑞芯微官方网站或GitHub仓库寻找最新的发布版本和支持文件。通常情况下,这些资源会包含详细的安装说明、兼容性列表以及其他重要信息[^2]。 #### 安装环境准备 在Ubuntu环境搭建RKNN-Toolkit-Lite2的工作流程与标准版相似,主要包括以下几个方面: - **下载官方项目包**:从GitHub克隆最新源码。 - **解决依赖项**:确保已安装TensorFlow、PyTorch等框架,并配置好编译器如GCC。 - **执行安装脚本**:按照README中的指示完成软件包的本地构建与安装过程。 #### 示例代码展示 下面是一个简单的Python示例,用于演示如何利用RKNN-Toolkit-Lite2加载预训练好的YOLOv5模型并进行推理操作: ```python import rknn.api as rknn_api def load_model_and_infer(image_path): # 初始化RKNN对象 rknn = rknn_api.RKNN() # 加载ONNX格式的YOLOv5模型 ret = rknn.load_onnx(model='./yolov5s.onnx') if ret != 0: print('Load YOLOv5 model failed!') exit(ret) # 构建计算图 ret = rknn.build(do_quantization=True, dataset='./dataset.txt') if ret != 0: print('Build model failed!') exit(ret) # 导出优化后的模型 rknn.export_rknn('./yolov5.rknn') # 设置输入数据 img = cv2.imread(image_path) img = preprocess_image(img) # 自定义图像前处理函数 # 执行推理 outputs = rknn.inference(inputs=[img]) # 后处理逻辑... ``` 此段代码展示了基本的应用场景,实际应用时可能还需要根据具体需求调整参数设置及前后处理部分[^4]。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值