1. 前提
用户得到了开发SDK包bmnnsdk2-bm1684和docker镜像BM1684-ubuntu-docker。
2. 安装lib和驱动
安装lib和驱动都是一次性的操作。
cd bmnnsdk2-bm1684_v2.2.0/scripts
2.1 安装lib
a) x86模式
./install_lib.sh nntc
b) arm pcie 模式
./install_lib.sh nntc cc_arm_pcie
2.2 安装Driver
a) x86模式
sudo ./install_driver_pcie.sh
b) arm pcie 模式
sudo ./install_driver_arm_pcie.sh
2.3 使用bm-smi查看驱动是否安装成功
bm-smi是经常被使用到的命令,可以用来查看板卡信息(tpu、内存占用率、温度等)。
a) x86模式
source envsetup_pcie.sh (这个命令主要是设置一些环境变量,所以退出终端后需要重新执行)
b) arm pcie 模式
source envsetup_arm_pcie.sh
然后执行
bm-smi
3. 加载docker镜像并进入docker
docker load -i BM1684-ubuntu-docker
cd bmnnsdk2-bm1684_v2.2.0
./docker_run_bmnnsdk.sh
执行后就进入了docker,可以在里面进行模型转换、量化、编译程序等操作。