Rockchip NPU 项目安装和配置指南
rknpu 项目地址: https://gitcode.com/gh_mirrors/rk/rknpu
1. 项目基础介绍和主要的编程语言
项目基础介绍
Rockchip NPU 项目(https://github.com/rockchip-linux/rknpu)是为 Rockchip 的 NPU(Neural Processing Unit)提供驱动和示例代码的开源项目。该项目适用于多个 Rockchip 平台,包括 RK1808、RK1806、RV1109、RV1126 等。通过该项目,开发者可以轻松地为这些平台编写和部署 AI 应用。
主要的编程语言
该项目主要使用 C 和 C++ 语言进行开发。此外,还涉及一些 Shell 脚本和 Python 脚本用于自动化和配置管理。
2. 项目使用的关键技术和框架
关键技术和框架
- RKNN Toolkit: 用于将原始模型转换为 RKNN 模型,支持联机调试和预编译功能。
- NPU 驱动: 提供 NPU 的驱动程序,包括 full driver 和 mini driver,支持不同的 Rockchip 平台。
- RGA(Rockchip Graphics Adapter): 用于图像处理的硬件加速器,常用于图像预处理。
3. 项目安装和配置的准备工作和详细的安装步骤
准备工作
- 硬件要求: 需要一块支持 Rockchip NPU 的开发板,如 RK1808、RK1806 等。
- 软件要求:
- 操作系统:Linux(推荐 Ubuntu 或 Debian)
- 开发工具:Git、CMake、GCC/G++ 编译器
- 其他工具:ADB(Android Debug Bridge)用于文件传输
详细的安装步骤
步骤 1:克隆项目仓库
首先,从 GitHub 克隆 Rockchip NPU 项目到本地:
git clone https://github.com/rockchip-linux/rknpu.git
cd rknpu
步骤 2:安装依赖
确保系统中安装了必要的依赖包:
sudo apt-get update
sudo apt-get install -y build-essential cmake git
步骤 3:编译驱动
进入驱动目录并编译驱动程序:
cd drivers
make
步骤 4:安装驱动
将编译好的驱动文件推送到开发板:
adb push drivers/linux-aarch64/* /
adb push drivers/npu_ko/galcore.ko /lib/modules/galcore.ko
步骤 5:配置环境
确保开发板上的 /usr/lib/npu/rknn/memory_profile
文件具有可执行权限:
adb shell
chmod +x /usr/lib/npu/rknn/memory_profile
步骤 6:测试安装
运行示例程序以验证安装是否成功:
cd examples
./run_example.sh
注意事项
- 如果遇到驱动加载失败或模型初始化问题,请检查驱动版本和权限设置,必要时联系开发团队获取支持。
- 对于不同的 Rockchip 平台,可能需要使用不同的驱动版本,请参考项目文档进行选择。
通过以上步骤,您应该能够成功安装和配置 Rockchip NPU 项目,并开始进行 AI 应用的开发和部署。