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,可以在里面进行模型转换、量化、编译程序等操作。

Logo

权威|前沿|技术|干货|国内首个API全生命周期开发者社区

更多推荐