开发前准备
环境准备
支持的产品型号:Atlas A2训练系列产品/Atlas 800I A2推理产品。
- 安装CANN软件后,使用CANN运行用户编译、运行时,需要以CANN运行用户登录环境,执行如下命令设置环境变量。其中${install_path}为CANN软件的安装目录,例如:root用户默认是"/usr/local/Ascend"。
上述环境变量只在当前窗口生效,用户可以将上述命令写入~/.bashrc文件,使其永久生效,方法如下:
- 以安装用户在任意目录下执行vi ~/.bashrc,在该文件最后添加上述内容。
- 执行:wq!命令保存文件并退出。
- 执行source ~/.bashrc使环境变量生效。
网络分析
通过FlowData和FlowNode构建Graph,需要根据网络,明确如下信息:
- 网络包含几个输入,通过FlowData表示。
- 网络包含几个计算节点,通过FlowNode表示。
- FlowNode里的实际计算由pp表示,pp有两种,FuncProcessPoint和GraphProcessPoint,如果是GraphProcessPoint,需要将原始模型保存为IR文件即可。如果是FuncProcessPoint参考UDF开发自定义用户功能。