RK3568 NPU RKNN(二):RKNN-ToolKit2环境搭建

发布于:2025-08-17 ⋅ 阅读:(21) ⋅ 点赞:(0)

1、前言

本文主要介绍如何在ubuntu下的搭建RKNN-ToolKit2开发环境。

2、下载安装Miniconda

Miniconda 是一个轻量级的 Conda 发行版,用于管理 Python 版本和虚拟环境。

2.1、下载miniconda

# 下载 Linux 64 位 Miniconda 最新版安装脚本
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh

2.2、安装miniconda

# 安装 Miniconda
bash Miniconda3-latest-Linux-x86_64.sh

安装完成后会自动设置环境变量。打开新的终端,发现用户名前出现(base),就代表安装成功:

3、下载安装RKNN-ToolKit2

RKNN-ToolKit2 开发套件运行在PC (x86_64/arm64)平台上,提供了模型转换、量化功能、模型推理、性能和内存评估、量化精度分析、模型加密等功能。

3.1、创建虚拟环境

# 创建一个名为rknn-toolkit2的虚拟环境,并指定python版本
conda create -n rknn-toolkit2 python=3.8
conda activate rknn-toolkit2

3.2、下载RKNN-ToolKit2

# 可以从官方地址获取:git clone https://github.com/airockchip/rknn-toolkit2.git
# 因为拉取速度太慢,这里使用野火提供的仓库
git clone https://gitee.com/LubanCat/lubancat_ai_manual_code.git

3.3、安装RKNN-ToolKit2安装依赖和包

# 进入rknn_toolkit2目录
cd lubancat_ai_manual_code/dev_env/rknn_toolkit2

# 配置pip源
pip3 config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple/

# 安装依赖库
pip3 install numpy
pip3 install -r doc/requirements_cp38-1.5.0.txt

# 安装rknn_toolkit2
pip3 install packages/rknn_toolkit2-1.5.0+1fa95b5c-cp38-cp38-linux_x86_64.whl

4、验证

(rknn-toolkit2) lubancat@lubancat-vm:~$ python3
Python 3.8.10 (default, May 26 2023, 14:05:08)
[GCC 9.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> from rknn.api import RKNN
>>>

5、总结

参考文章:

https://doc.embedfire.com/linux/rk356x/Ai/zh/latest/lubancat_ai/env/toolkit2.html#


网站公告

今日签到

点亮在社区的每一天
去签到