前言:本篇文章是介绍寒武纪模型移植前的一些准备工作
一、环境搭建
硬件环境:mlu220芯片、ubuntu18.04_arm64
开发环境:Windows10、VMware
1.1 简介
寒武纪本身支持多个框架的模型导出,但是由于和Pytroch等框架深度绑定,因此不同框架需要安装不同的模型导出环境。
博主手上有【pytorch-0.11.118-2-ubuntu16.04.tar】【pytorch-0.15.614-2-ubuntu18.04.tar.gz】两个开发环境,经过测试,0.15会更加稳定一点,优先用最新的版本。
如果需要实际开发产品,需要基于寒武纪的CNStream【cnstream_v1.0.tar】搭建运行环境。单独运行模型则可以直接在ubuntu18.04_arm64的基础上运行。
1.2 导入
博主喜欢用Windows + VMware + ubuntu18.04 + VScode + Samba的形式搭建开发环境,使用Podman作为镜像管理工具。
使用Podman导入完Pytorch镜像后,可以通过SDK包中resnet来学习一下模型转换常规流程。
二、名词解释
这块是比较烦的,每一个新的框架都会更新许多新的名词,这里就大致介绍一下:
CNRT - 运行时库(支持MLU200系列)
CNML - 机器学习算子库
BANG C - MLU编程语言, 类似英伟达里的 CUDA C
CNCC - 编译 BANG C 程序的编译器
MagicMind - 运行时库(支持MLU200/MLU300系列)
CNCV - 预处理模块
三、资源下载
还没有想好什么资源是可以放上来的,不过大家应该都有的吧。。。