华为部署生态
-
CANN:华为昇腾的异构计算架构
-
Ascend:基于达芬奇架构的计算卡,昇腾310,昇腾910等。
-
MindSpore:基于昇腾架构研发的计算框架,类似于nvidia的TensorRT。
-
AscendCL:基于昇腾架构的计算语言,类似于nvidia的CUDA语言。
华为昇腾开发环境
首先有了硬件之后,需要先装驱动。驱动装完之后往上需要装用于推理的软件包,叫nnrt
(Nerual Network RunTime
)。这样我们就已经装完了运行的环境。为了我们开发方便,我们还需要在其之上装一个Toolkit
。这些和nvidia都是非常像的,驱动 -> nvcc -> toolkit。
完成上述步骤,就可以在命令行编译运行我们的程序了。如果为了方便开发,还可以再安装一个MindStudio
的软件。
驱动安装
首先关注一下华为昇腾社区的官网:https://www.hiascend.com
点击产品
中的固件和驱动
,选择加速卡
的产品系列,由于此次公司的计算卡是昇腾910处理器
,这里选择Atlas 300T Pro训练卡(型号:9000)
。选择.run
格式的驱动包,自行选择ARM
和X86_64
架构的包。需要注意的是这里选择的CANN版本与后面要安装的nnrt版本有关联。
CANN架构包下载
点击产品
中的CANN
,选择社区版下载,选择与驱动安装时相同的架构的选项,选择nnrt
和toolkit
的包进行下载。
nnrt
是推理包,nnae
是面向训练的包。
MindStudio下载
MindStudio
提供您在AI开发所需的一站式开发环境,支持模型开发、算子开发以及应用开发三个主流程中的开发任务。
依靠模型可视化、算力测试、IDE本地仿真调试等功能,MindStudio能够帮助您在一个工具上就能高效便捷地完成AI应用开发
MindStudio采用了插件化扩展机制,开发者可以通过开发插件来扩展已有功能。
选择产品
中的全流程开发工具链MindStudio
,选择对应操作系统的软件包进行下载。
开发者文档
点击右上角的文档
选项,进入昇腾文档,下滑,找到如下页面:
点击CANN软件安装
,选择安装开发环境
,按照文档进行对应系统的依赖。
然后按照安装驱动(包含重启)
->安装nnrt(包含配置环境变量)
->安装toolkit
->安装MindStudio
:
|
|
安装完软件之后,为软件配置一下toolkit源文件,界面类似于PyCharm
,比较重要的就是模型转化的过程了。
MindStudio模型推理
- 模型转换:点击软件的
Ascend
,再选择Model Converter
,选择模型文件,然后选择精度为FP16
,等待转换完成。
在工程目录下,点击右键,Add model
,选择转换好的.om
文件。.om
文件的地址一般在你的用户名的文件夹下的modelzoo
文件夹。
- 图片转换:对准备好的图片放入
data
文件夹下,并进行转换:
|
|
- 编译:点击
Build
目录下的Edit Configurations
,然后直接点击build。 - 运行:build成功之后在
out
文件夹下会出现一个main
的可执行文件。点击Run
目录下的Edit Configurations
,选择Local Run
,并选择可执行文件main
。然后再次点击Run
目录下Run MyApp(你的工程名)
,就开始执行了。
最后修改于 2023-08-09
本作品采用知识共享署名-非商业性使用-相同方式共享 4.0 国际许可协议进行许可。