昇腾社区首页
中文
注册
开发者
下载

环境配置

如果需要使用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,用户进程结束后自动失效。

表1 环境变量列表

环境变量

说明

LD_LIBRARY_PATH

动态库的查找路径。

ASCEND_CUSTOM_OPP_PATH

推理引擎自定义算子包安装路径。

ASCEND_RT_VISIBLE_DEVICES

指定当前进程所用的昇腾AI处理器的逻辑ID,如有需要请自行配置。

配置示例:"0,1,2"或"0-2";昇腾AI处理器的逻辑ID间使用“,”表示分割,使用“-”表示连续。