比特大陆/算能科技BM1684芯片环境配置

引子

　　国产AI芯片一直是备受大家关注的一个方面，刚好工作中使用过，在此，要做一个简单的总结和汇总，OK，让我们开始吧。PS：最近准备对国产AI芯片做一个专辑，敬请期待。

一、环境安装

1、下载镜像

技术资料

2、下载sdk2.7

百度网盘 请输入提取码

提取码：nzpx

3、解压sdk

tar zxvf bmnnsdk2-bm1684_v2.7.0

cd bmnnsdk2-bm1684_v2.7.0

4、验证驱动

lspci | grep 1684

正常输出如下：

01:00.0 Processing accelerators: Bitmain Technologies Inc. BM1684, Sophon Series Deep Learning Accelerator (rev 01)

5、启动docker

从宿主机SDK根目录下执行脚本进入docker环境

./docker_run_bmnnsdk.sh

6、配置环境

cd /workspace/scripts/

./install_lib.sh nntc

source envsetup_pcie.sh

export PYTHONPATH=$PYTHONPATH:/workspace/lib/opencv/x86/opencv-python

7、验证

cd /workspace/examples/SSD_object/cpp_cv_bmcv_bmrt/

make -f Makefile.pcie clean && make -f Makefile.pcie

不报错即为安装完成

二、跑通SSD官方例程C++

1、下载模型和数据

（1）下载

cd /workspace/examples/SSD_object/model

./download_ssd_model.sh

（注:若您无法访问google网络硬盘，请从以下百度网盘链接下载后拷贝到该目录。

原始Caffe模型数据：models_VGGNet_VOC0712_SSD_300x300.tar.gz:

百度网盘 请输入提取码 Password: i4x9

量化使用的图片集数据：VOC712.tgz:

百度网盘 请输入提取码 Password：nl7v）

（2）解压所需文件

切换到指定目录

cd ${BMNNSDK}/examples/SSD_object/model

解压数据集到data目录下，完成后应该能看到data.mdb lock.mdb

tar zxf VOC712.tgz全路径 -C ./data/

将下载的文件models_VGGNet_VOC0712_SSD_300x300.tar.gz放到examples/SSD_object/model目录下

cp models_VGGNet_VOC0712_SSD_300x300.tar.gz examples/SSD_object/model

2、生成fp32 bmodel

进入docker（同一中5、6步骤）

（1）切换到指定目录

cd /workspace/examples/SSD_object/model

（2） 生成fp32 bmodel

./gen_bmodel.sh

（3）fp32模型转换完成之后，接着执行如下指令生成int8 bmodel

./gen_umodel_int8bmodel.sh

以上步骤正确执行后，会在out目录下生成fp32_ssd300.bmodel 、int8_ssd300.bmodel文件。

3、程序编译、运行

cd /workspace/examples/SSD_object/cpp_cv_bmcv_bmrt

make -f Makefile.pcie # for PCIE MODE

make -f Makefile.arm # for SoC MODE

PCIE模式下运行

./ssd300_cv_bmcv_bmrt.pcie image car.jpg ../model/out/fp32_ssd300.bmodel 1 0

运行后，会在当前目录创建results文件夹保存检测结果。

三、python版本例程

1、安装sophon python sdk

cd /workspace/lib/sail/python3/pcie/py37

pip3 install sophon-2.7.0-py3-none-any.whl

2、下载模型

cd /workspace/examples/sail/sc5_tests/

python3 ./download.py ssd_fp32.bmodel

3、运行模型

(1)ssd

mv ssd_fp32_191115.bmodel python/det_ssd/

cd python/det_ssd

python3 det_ssd_0.py --bmodel ./ssd_fp32_191115.bmodel --input ./car.jpg

(2)yolov5s

cd /workspace/examples/YOLOv5_object/python

python3 sail.py --img-name car.jpg