rknn.list_devices()
是 RKNN-Toolkit2 中用于 获取当前连接的 NPU 设备列表 的接口,帮助开发者识别和管理多个硬件设备(如 RK3588、RK3566 开发板)。以下是其功能、使用示例及常见问题的完整说明:
一、功能与作用
- 核心功能:
- 列出通过 USB、PCIe 或 ADB 连接到主机的 Rockchip NPU 设备。
- 返回设备 ID 列表,用于后续
init_runtime()
指定目标设备。
- 适用场景:
- 多设备环境下选择特定 NPU 进行模型部署。
- 验证设备驱动是否安装成功,排查硬件连接问题。
二、参数与返回值
- 参数:无。
- 返回值:
返回一个 字符串列表,格式取决于连接方式:- USB/PCIe 直连:设备 ID 为
usb://0
、pcie://0
等。
- USB/PCIe 直连:设备 ID 为