文档
注册

安装前准备

量化并转换MindSpore模型前,请先完成如下环境准备动作,然后安装AMCT

用户准备

请使用训练环境下CANN软件包的安装用户安装AMCT,本章节以非root用户为例进行操作。

关于非root用户权限请参见CANN软件安装指南手册>安装开发环境>准备安装及运行用户章节。

环境准备

AMCT需要在带NPU设备的训练环境下安装,配套关系如下:

表1 配套版本信息

类别

版本限制

获取方式

注意事项

带NPU设备的训练环境所需操作系统

请参见支持的OS清单

参见CANN软件安装指南手册搭建对应环境。

仅支持基于NPU的量化。

MindSpore

1.10.0

https://www.mindspore.cn/install

安装完后,需要配置环境变量,并建议进行安装验证。

Python

Python3.7.x(3.7.5~3.7.11)、Python3.8.x(3.8.0~3.8.11)、Python3.9.x(3.9.0~3.9.2)

推荐使用Python3.9.2

AMCT以及训练环境搭建都需要安装该依赖,具体安装方法请参见CANN软件安装指南手册>安装开发环境>安装依赖章节。

若需要搭建Python3.9.2环境,则:

  • 安装依赖时,请确保服务器能够连接网络。
  • 本手册以Python3.9.2为例进行介绍,相应环境变量和安装命令以实际安装Python版本为准。

numpy

1.16.0+

请参见安装依赖

-

protobuf

3.13.0+

-

安装依赖

请参见CANN软件安装指南>安装开发环境>安装依赖章节。安装完成后,还需要执行如下命令安装相关依赖。

执行pip3 list命令检查wget、numpy、protobuf 软件是否安装,若提示相应版本信息,则说明已经安装,否则请执行如下安装命令。

表2 依赖列表

依赖名称

版本号

安装命令

numpy

1.16.0+

pip3 install numpy==1.16.0 --user

protobuf

3.13.0+

pip3 install protobuf==3.13.0 --user

wget

3.2

pip3 install wget==3.2 --user

上传软件包

AMCT的安装用户将Ascend-cann-amct_{software version}_linux-{arch}.tar.gz软件包上传到Linux服务器任意目录下,本示例为上传到$HOME/amct/目录。

执行如下命令解压AMCT软件包:

tar -zxvf Ascend-cann-amct-{software version}_linux-{arch}.tar.gz

获得如下内容:

表3 AMCT软件包解压后内容

一级目录

二级目录

说明

使用场景及注意事项

amct_caffe/

Caffe框架AMCT目录。

  • 使用方法请参见AMCT工具(Caffe)
  • 需要搭建Caffe环境,量化完的模型,如果要执行推理,则需要借助安装昇腾AI处理器的推理环境。

amct_caffe-{version}-py3-none-linux_{arch}.whl

Caffe框架AMCT安装包。

caffe_patch.tar.gz

Caffe源代码增强包。

amct_tf/

TensorFlow框架AMCT目录。

amct_tensorflow-{version}-py3-none-linux_{arch}.tar.gz

TensorFlow框架AMCT安装包,通过源码编译方式进行安装。

  • amct_tensorflow与amct_tensorflow_ascend软件包不能同时安装
  • 使用方法请参见AMCT工具(TensorFlow)
  • 需要搭建TensorFlow环境,量化完的模型,如果要执行推理,则需要借助安装昇腾AI处理器的推理环境。

amct_tensorflow_ascend-{version}-py3-none-linux_{arch}.tar.gz

基于TF_Adapter的AMCT安装包,通过源码编译方式进行安装。

  • amct_tensorflow与amct_tensorflow_ascend软件包不能同时安装。
  • Atlas 200/300/500 推理产品:不支持amct_tensorflow_ascend目录下的相关特性。
  • Atlas 200/500 A2推理产品:不支持amct_tensorflow_ascend目录下的相关特性。
  • Atlas 推理系列产品(Ascend 310P处理器):不支持amct_tensorflow_ascend目录下的相关特性。
  • 使用方法请参见AMCT工具(TensorFlow,Ascend)
  • 需要搭建TensorFlow环境,并借助带有NPU设备的在线推理环境,量化完的模型,可以基于昇腾AI处理器完成在线推理业务。

amct_pytorch/

PyTorch框架AMCT目录。

  • 使用方法请参见AMCT工具(PyTorch)
  • 需要搭建PyTorch环境,量化完的模型,如果要执行推理,则需要借助安装昇腾AI处理器的推理环境。

amct_pytorch-{version}-py3-none-linux_{arch}.tar.gz

PyTorch框架AMCT源码安装包。

amct_onnx/

ONNX模型AMCT目录。

amct_onnx-{version}-py3-none-linux_{arch}.whl

ONNX模型AMCT安装包。

amct_onnx_op.tar.gz

AMCT基于ONNX Runtime的自定义算子包。

amct_ms/

MindSpore框架AMCT目录。

  • Atlas 200/300/500 推理产品:不支持amct_ms目录下的相关特性。
  • Atlas 200/500 A2推理产品:不支持amct_ms目录下的相关特性。
  • Atlas 推理系列产品(Ascend 310P处理器):不支持amct_ms目录下的相关特性。
  • 使用方法请参见AMCT工具(MindSpore)
  • 需要搭建MindSpore环境,并借助带NPU设备的训练环境将MindSpore模型量化后输出为*.air格式模型,只能借助带NPU设备的训练环境,量化后的模型要进行推理,也可以借助该环境完成。

amct_mindspore-{version}-py3-none-linux_{arch}.whl

MindSpore框架AMCT安装包。

third_party/

Ascend-nca-{software version}-linux.{arch}.run

NCA(Neural Compute Agent)软件包。

基于性能的自动量化场景下使用。该版本不支持third_party目录下的相关特性。

其中:{version}表示AMCT具体版本号,{software version}为软件版本号,{arch}表示具体操作系统架构。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词