昇腾社区首页
中文
注册

AMCT(MindSpore)

安装完AMCT工具,执行量化操作前,需要参见该章节设置相关环境变量。

  • CANN软件包安装完成后,执行量化前,需要设置如下环境变量,请以CANN软件包的运行用户执行如下命令生效环境变量,其中${install_path}为CANN软件包安装路径,请根据实际情况进行替换:
    # 昇腾设备安装部署开发套件包Ascend-cann-toolkit。
    . ${install_path}/ascend-toolkit/set_env.sh 
    # 昇腾设备安装部署软件包Ascend-cann-nnae。
    . ${install_path}/nnae/set_env.sh
  • 设置生成权重和数据量化因子的环境变量,调用save_model时会在当前运行脚本目录中新建amct_dump目录,并在该目录生成record.txt文件,用于记录权重和数据的量化因子,如果要进行量化感知训练,则环境变量必须设置:
    export DUMP_AMCT_RECORD=1

用户也可以通过修改~/.bashrc文件方式设置永久环境变量,如下以bash shell为例进行说明:

  1. 以安装用户在任意目录下执行vi ~/.bashrc命令,打开.bashrc文件,在文件最后一行后面添加上述内容。
  2. 执行:wq!命令保存文件并退出。
  3. 执行source ~/.bashrc命令使其立即生效。