1. 环境准备:从零搭建昇腾开发环境
第一次拿到Atlas 200I DK A2开发板时,我对着这个巴掌大的黑色盒子研究了半天。别看它体积小,里面可是藏着华为昇腾310B1芯片,NPU算力高达8TOPS(INT8)。咱们部署YOLOv5的第一步,就是给这个"小钢炮"搭建好开发环境。
开发板出厂时已经预装了Ubuntu 18.04系统,但有几个关键组件需要确认。用网线连接开发板后,通过ssh登录(默认IP是192.168.1.2),先运行npu-smi info命令。如果看到类似下面的输出,说明NPU驱动正常:
bash复制+------------------------------------------------------------------------------+
| npu-smi 21.0.4 Version: 21.0.4 |
+----------------------+---------------+----------------------------------------+
| NPU Name | Health | Power(W) Temp(C) |
| Chip | Bus-Id | AICore(%) Memory-Usage(MB) |
+======================+===============+========================================+
| 0 310B1 | OK | 5.4 45 |
| 0 | 0000:01:00.0 | 0 224/7972 |
+======================+===============+========================================+
接下来要安装CANN工具包(当前推荐5.1.RC2版本),这个相当于NPU的"驱动程序"。我建议直接使用华为提供的Docker镜像,省去一堆依赖问题:
bash复制# 拉取官方镜像
docker pull swr.cn-south-1.myhuaweicloud.com/ascend-share/ascend_ubuntu18.04_aarch64:5.1.RC2
# 启动
