跳到主要内容

NPU开发环境搭建

本章介绍如何在计算机中直接安装开发环境,后续的示例程序运行过程也将以直接安装为例说明。如果需要在 Docker 环境中运行示例程序,可以参考第 5 章中的内容准备开发环境。

本章分为四个部分:

  • 下载 RKNN 相关仓库
  • 安装RKNN-Toolkit2 环境
  • 安装编译工具
  • 安装板端 RKNPU2 环境

1 下载 RKNN 相关仓库
建议新建一个目录用来存放 RKNN 仓库,例如新建一个名称为rknn_prj的文件夹,并将 RKNN-Toolkit2和 RKNN Model Zoo 仓库存放至该目录下,参考命令如下:

2 安装 RKNN-Toolkit2 环境

2.1 安装 Python
如果系统中没有安装 Python 3.8(建议版本),或者同时有多个版本的 Python 环境,建议使用Miniforge Conda或virtualenv创建新的 Python 3.8 环境来隔离。

2.1.1 安装 Miniforge Conda
在计算机的终端窗口中执行以下命令,检查是否安装Miniforge Conda,若已安装则可省略此节步骤。

如果没有安装 Miniforge Conda,可以通过下面的链接下载 Miniforge Conda 安装包并安装:

2.1.2 使用 Miniforge Conda 创建 Python 环境

2.2 安装 RKNN-Toolkit2
激活 toolkit2 环境后,可通过 pip 源或者本地 wheel 包这两种方式安装 RKNN-Toolkit2:

通过 pip 源安装

通过本地 wheel 包安装

是处理器架构类型 (x86_64 对应
x86_64/aarch64 对应 ARM64)
pip install rknn_prj/rknn_toolkit2--x.x.x-cpxx-cpxx-manylinux_2_17_<arch>.manylinux2014_<arch>.whl]]></arch></arch>

2.3 验证是否安装成功
执行以下命令,若没有报错,则代表 RKNN-Toolkit2 环境安装成功。

如果安装失败,请查阅 SDK《Rockchip_RKNPU_User_Guide_RKNN_SDK_CN.pdf》文档中的第 10.2 章节 “工具安装问题”,其中详细介绍了 RKNN-Toolkit2 环境安装失败的解决方法。

3.3 电脑端安装编译工具

3.3.1 安装 CMake,在计算机的终端中,执行以下命令:

3.2 安装交叉编译器
3.2.1Android 系统开发板安装 NDK

注:该章节适用于 Android 系统的开发板,如果板端是 Linux 系统,请忽略此章节。

此时,NDK 编译器的路径就是 rknn_prj/android-ndk-r19c

3.2.2 Linux 系统开发板安装 GCC 交叉编译器
注:该章节适用于 Linux 系统的开发板,如果板端是 Android 系统,请忽略此章节。
GCC 下载地址
板端为 64 位系统:https://releases.linaro.org/components/toolchain/binaries/6.3-2017.05/aarch64-linux-gnu/gcc-linaro-6.3.1-2017.05-x86_64_aarch64-linux-gnu.tar.xz
板端为 32 位系统:https://releases.linaro.org/components/toolchain/binaries/6.3-2017.05/arm-linuxgnueabihf/gcc-linaro-6.3.1-2017.05-x86_64_arm-linux-gnueabihf.tar.xz
解压软件包
建议将 GCC 软件包解压到 rknn_prj 的文件夹中。以板端为 64 位系统的 GCC 软件包为例,存放位置如下:

此时, GCC 编译器的路径是 Projects/gcc-linaro-6.3.1-2017.05-x86_64_aarch64-linux-gnu/bin/aarch64-linux-gnu

3.4 安装板端 RKNPU2 环境
3.4.1 确认 RKNPU2 驱动版本
可以在板端执行以下命令查询 RKNPU2 驱动版本:

3.4.2 检查 RKNPU2 环境是否安装