迅为RK3576开发板NPU例程测试-rknn-toolkit2环境搭建和使用

发布于:2025-06-18 ⋅ 阅读:(21) ⋅ 点赞:(0)

迅为iTOP-3576开发板采用瑞芯微RK3576高性能、低功耗的应用处理芯片,集成了4个Cortex-A72和4个Cortex-A53核心,以及独立的NEON协处理器。它适用于ARM PC、边缘计算、个人移动互联网设备及其他多媒体产品。

2.1 rknn-toolkit2 环境搭建

rknn-toolkit2 的环境搭建实际上就是安装瑞芯微提供的 rknn-toolkit2 whl 文件,但由于各个Linux环境有着不同的Python版本,所以RK也提供了不同版本的安装包(虽然RK也提供了arm64版本的 whl,可以在开发板上进行模型转换,但由于开发板性能远远比不上 PC 端的性能,所以只讲解在 PC 端进行模型转换的方法),具体如下图所示:

而根据不同的场景提供了三种 knn-toolkit2 的安装方法,分别为直接安装、conda 虚拟环境安装以及 docker 环境安装,直接安装的方法最为省时省力,建议大家使用第一种方法,但直接安装会更改当前的环境,如果不想更改当前的环境可以使用第二种方法 conda 虚拟环境安装,而 docker 环境安装为瑞芯微已经提供了一个安装好 knn-toolkit2 的 ubuntu20 docker 环境,加载 docker 环境后就可直接使用,每种方法有各自的优点和缺点,大家可以根据自己的情况 选择对应的安装方法。

2.1.1 直接安装

瑞芯微提供了六个 Python 版本的的 knn-toolkit2 whl 安装包,可以根据自己的 Linux 环境来选择合适版本的安装包,具体的对应关系如下所示:

这里以 ubuntu20 为例进行 knn-toolkit2 的安装,ubuntu20 对应的 python 版本为 3.8,具体如下所示:

然后来到 rknn-toolkit2/rknn-toolkit2/packages/X86_64 目录下,如下图所示:

这里要用到的为 Python3.8 的依赖项文件以及 rknn_toolkit2 whl 文件,具体如下所示:

首先使用以下命令安装依赖项,如下图所示:

pip install -r requirements_cp38-2.3.0.txt -i https://repo.huaweicloud.com/repository/pypi/simple/

中间会遇到一些警告忽略即可,

然后使用以下命令安装 rknn_toolkit2 whl 文件,安装过程如下图所示:

pip install rknn_toolkit2-2.3.0-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl -i https://repo.huaweicloud.com/repository/pypi/simple/

安装完成之后输入 python3 进入 python 命令行,然后依次输入以下内容,能打印出 rknn_

toolkit2 版本证明安装成功,如下图所示:

from rknn.api import RKNN

RKNN(verbose=True)

至此,直接安装 rknn_toolkit2 就演示完成了,在后面的小节中会对 rknn_toolkit2 的实际功能进行演示。

2.1.2 conda 虚拟环境安装本小节要用到的软件安装包路径为“iTOP-3576 开发板\02_【iTOP-RK3576 开发板】开发资料\05_NPU 例程测试配套资料\03_miniconda”

在上个小节中讲解了在 Linux 环境中直接安装 rknn_toolkit2 的方法,但这种方法可能会影响本地的环境,将之前配置的一些环境依赖搞乱,为了解决这种问题可以使用 conda 创建一个 rknn_toolkit2 专属 python 环境,从而实现不同环境之间的隔离问题。

conda 可以直接在 https://repo.anaconda.com/miniconda/网站进行下载,也可以之间使用迅为已经下载好的安装包,最后将它拷贝到 Linxu 环境中,拷贝完成如下图所示:

然后使用 chmod 命令赋予该文件可执行权限,然后运行,

根据步骤进行安装即可,安装完成之后重新打开一个终端,或者使用下面的 source 命令重新加载环境变量,就会在终端的最前面多出 base 的字符,

source ~/.bashrc

最前方为 bash 表示处在默认的配置环境,这时候需要创建 rknn_toolkit2 专属 python 环境, 可以使用以下命令创建一个名为 rknn 的、Python 版本为 3.8 的虚拟环境,注:这里的 Python 版本可以任意设置,只需要选择同版本的 rknn_toolkit2 whl 包即可。

conda create -n rknn python=3.8

虚拟环境创建完成如下图所示:

然后使用“conda activate rknn”命令激活刚刚创建完成的 rknn 环境,激活完成之后终端最前方的 base 字符会更改为 rknn,

然后来到 rknn-toolkit2/rknn-toolkit2/packages/x86_64 目录下,

由于前面创建的是 Python 版本为 3.8 的环境,所以这里要用到的为 Python3.8 的依赖项文件以及 rknn_toolkit2 whl 文件,具体如下所示:

首先使用以下命令安装依赖项,如下图所示:

pip install -r requirements_cp38-2.3.0.txt -i https://repo.huaweicloud.com/repository/pypi/simple/

安装完成如下图所示:

然后使用以下命令安装 rknn_toolkit2 whl 文件,安装过程如下图所示:

pip install

rknn_toolkit2-2.3.0-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

-i

https://repo.huaweicloud.com/repository/pypi/simple/

安装完成之后输入 python3 进入 python 命令行,然后依次输入以下内容,能打印出 rknn_ toolkit2 版本证明安装成功,如下图所示:

from rknn.api import RKNN

RKNN(verbose=True)

至此,使用 conda 虚拟环境安装 rknn_toolkit2 就演示完成了,在后面的小节中会对 rknn_t oolkit2 的实际功能进行演示。


网站公告

今日签到

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