BMNNSDK版本特性
BMNNSDK 包含设备驱动、运行库、头文件和相应工具,主要特性如下:
1)设备驱动
PCIE 支持多种 Linux 发行版本和 Linux 内核。
SOC 模式提供 ko 模块,可以直接安装到BM168x SOC Linux Release。
CModel 模式,使用 CPU 模拟 Sophon AI hardware,开发者可以先使用 cmodel 验证软件, 而不需要依赖Sophon Hardware 。
2)运行库
提供深度学习推理引擎,提供最大的推理吞吐量和最简单的应用部署环境。
提供三层接口,网络级接口/layer 级接口/指令级接口 。
利用TPU 提供灵活的图像处理功能(BMCV) • 提供运行库编程接口,用户可以直接操作bmlib/bmdnn 等底层接口,进行深度的开发 。
运行库支持多线程、多进程,提供并发处理能力。
3)工具
提供 bmnetc 工具,支持 Caffe 网络模型进行编译 。
提供 bmnett 工具, 支持 TensorFlow 模型进行编译。
提供 bmnetm 工具,支持 MxNet 模型进行编译 。
提供 bmnetp 工具,支持 PyTorch 模型进行编译 。
提供 bm_model.bin 工具,查看 bmodel 文件的参数信息,也可以将 bmodel 文件进行分解和合并。
提供 profiling 的工具,展示每一层执行所使用的指令以及指令所消耗的时间 。
提供 bm_smi 工具,支持对 BM1682 设备的运行状态进行监测。
Last updated
Was this helpful?