141 lines
4.0 KiB
Markdown
141 lines
4.0 KiB
Markdown
# RKNPU
|
||
|
||
本工程主要为Rockchip NPU提供驱动、示例等。
|
||
|
||
|
||
|
||
## 适用平台
|
||
|
||
- RK1808/RK1806
|
||
|
||
- RV1109/RV1126
|
||
|
||
|
||
|
||
**注意：**
|
||
|
||
- **RK3399Pro用户态的库及驱动不在本工程**，请参考：https://github.com/airockchip/RK3399Pro_npu
|
||
- RK3566/RK3568/RK3588/RV1103/RV1106请参考：https://github.com/rockchip-linux/rknpu2
|
||
|
||
|
||
|
||
## RKNN Toolkit
|
||
|
||
在使用RKNN API进行部署之前，需要使用RKNN Toolkit将原始的模型转化成rknn模型。
|
||
|
||
- RK1808/RK1806/RV1109/RV1126/RK3399Pro 使用： https://github.com/rockchip-linux/rknn-toolkit
|
||
- RK3566/RK3568/RK3588/RV1103/RV1106使用：https://github.com/rockchip-linux/rknn-toolkit2
|
||
|
||
具体的使用说明请参考相应的网址。
|
||
|
||
|
||
|
||
## NPU驱动说明
|
||
|
||
### NPU驱动目录说明
|
||
|
||
NPU的驱动在$SDK/external/rknpu/drivers/目录下或者https://github.com/rockchip-linux/rknpu/tree/master/drivers
|
||
|
||
其中的编译、安装规则参考$SDK/buildroot/package/rockchip/rknpu/rknpu.mk
|
||
|
||
主要目录包括：
|
||
|
||
```
|
||
drivers/
|
||
├── common
|
||
├── linux-aarch64
|
||
├── linux-aarch64-mini
|
||
├── linux-armhf
|
||
├── linux-armhf-mini
|
||
├── linux-armhf-puma
|
||
├── linux-armhf-puma-mini
|
||
├── npu_ko
|
||
```
|
||
|
||
- linux-aarch64：RK1808 full driver
|
||
|
||
- linux-aarch64-mini: RK1808 mini driver
|
||
|
||
- linux-armhf: RK1806 full driver
|
||
|
||
- linux-armhf-mini: RK1806 mini driver
|
||
|
||
- linux-armhf-puma: RV1109/RV1126 full driver
|
||
|
||
- linux-armhf-puma-mini: RV1109/RV1126 mini driver
|
||
|
||
- npu_ko：NPU内核驱动KO
|
||
|
||
在更新驱动时，需要同时更新用户态驱动及内核驱动，不然会产生类似错误:
|
||
|
||
```
|
||
[ 1] HAL user version 6.4.6.5.351518
|
||
[ 2] HAL kernel version 6.4.6.5.351518
|
||
```
|
||
|
||
另外，npu ko与内核配置强相关，有可能会加载不成功，运行程序时，会产生类似错误：
|
||
|
||
```
|
||
[ 1] Failed to open device: No such file or directory, Try again...
|
||
[ 2] Failed to open device: No such file or directory, Try again...
|
||
```
|
||
|
||
遇到这种情况需要联系开发团队解决。
|
||
|
||
|
||
|
||
### NPU full driver与mini driver的区别
|
||
|
||
主要包含以下几点：
|
||
|
||
- Mini driver只支持预编译的rknn模型，如果跑非预编译模型，会出现 RKNN_ERR_MODEL_INVALID的错误，从1.6.0开始，会返回RKNN_ERR_NON_PRE_COMPILED_MODEL_ON_MINI_DRIVER的错误;
|
||
|
||
- Full driver支持RKNN Toolkit的联机调试功能，mini driver不支持；
|
||
|
||
- Mini driver 库大小比full driver小很多，以RV1109/RV1126 1.6.0驱动为例，full driver大小为87MB，mini driver大小为7.1MB，可以有效的节省flash大小。
|
||
|
||
- Mini driver库运行时占用的内存比full driver小。
|
||
|
||
|
||
|
||
### 手动更新NPU驱动
|
||
|
||
有时需要手动更新NPU驱动，只要将相关的驱动拷贝到对应的目录就可以。
|
||
|
||
- 更新RK1808的驱动：
|
||
|
||
```
|
||
adb push drivers/linux-aarch64/ /
|
||
adb push drivers/npu_ko/galcore.ko /lib/modules/galcore.ko
|
||
```
|
||
|
||
- 更新RK1808 mini driver：
|
||
|
||
```
|
||
adb push drivers/linux-aarch64-mini/ /
|
||
adb push drivers/npu_ko/galcore.ko /lib/modules/galcore.ko
|
||
```
|
||
- 更新RV1109/RV1126的驱动：
|
||
|
||
```
|
||
adb push drivers/linux-armhf-puma/ /
|
||
adb push drivers/npu_ko/galcore_puma.ko /lib/modules/galcore.ko
|
||
```
|
||
|
||
- 更新RV1109/RV1126 mini driver：
|
||
|
||
```
|
||
adb push drivers/linux-armhf-puma-mini/ /
|
||
adb push drivers/npu_ko/galcore_puma.ko /lib/modules/galcore.ko
|
||
```
|
||
|
||
注意：adb不同版本的默认行为不一样，有些是直接push目录的，有些是push目录下的文件。因此替换完后，需要检查一下相关的库是否已经替换成功。比如通过md5sum检查对应文件的md5值是否相同。部分板子不支持adb命令，请通过ssh等其他方式拷贝文件到对应目录即可。
|
||
|
||
**另外，toybrick可能对文件路径进行了调整，因此最好通过toybrick自带的方式进行更新。**
|
||
|
||
|
||
|
||
## librknn_api与librknn_runtime的区别
|
||
|
||
librknn_api是对librknn_runtime的封装，主要是为了减少对其他so的编译依赖，功能上并没有区别。检查驱动版本时，一般以librknn_runtime.so为准。
|