使用描述
1、新购的香橙派5-rk3588s
2、有3个npu,号称6TOPS的算力
3、安装了linux6的Linux6 / 22.04 / desktop
【顺手提供镜像下载地址 https://openboard.dev/download/orangepi/5】
4、 写一个python/C++代码来跑yolov8进行推理
代码如下
rknn_lite = RKNNLite(verbose=True)
ret = rknn_lite.load_rknn(rknnModel)
if ret != 0:
print("Load RKNN rknnModel failed")
exit(ret)
# 开多个线程,让系统自动选择NPU去工作
# 或者每个线程用一个不同的npu去工作
if id == 0:
#ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_AUTO)# verbose=True
ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_0)
elif id == 1:
#ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_AUTO)
ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_1)
elif id == 2:
#ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_AUTO)
ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_2)
elif id == -1:
ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_0_1_2)
else:
ret = rknn_lite.init_runtime()
if ret != 0:
print("Init runtime environment failed")
exit(ret)
print(rknnModel, "\t\tdone")
问题描述
只有NPU_CORE_0是正常推理工作,NPU_CORE_1和NPU_CORE_2都无法进行推理工作
折腾了好多天,最后发现是入了官方的坑了,下错系统了。官方太可恶了,不能用的系统还提供给用户下载,而且没有任何的问题说明。
解决的方案:
更换到下面的系统
Linux5 / 22.04 / desktop | 1.1.8 / MD5 |
Linux5 / 22.04 / server | 1.1.8 / MD5 |
Linux5 / 20.04 / desktop | 1.1.8 / MD5 |
Linux5 / 20.04 / server | 1.1.8 / MD5 |
效果如下