环境配置
如果需要使用MindIE SD量化功能时,则必须安装CANN软件包NNAL神经网络加速库的Python库,即安装CANN软件包NNAL神经网络加速库时需添加--torch_atb参数进行指定,且运行时确保已安装ops算子包。安装命令如下所示:
./Ascend-cann-nnal_<version>_linux-<arch>.run --install --torch_atb
配置环境变量
MindIE安装完成后,提供进程级环境变量设置脚本“set_env.sh”以自动完成环境变量设置,该环境变量设置脚本包含如表1所示的ASCENDIE_HOME、LD_LIBRARY_PATH和ASCEND_CUSTOM_OPP_PATH,用户进程结束后自动失效。