昇腾社区首页
中文
注册

安装与部署

准备开发环境

MindIE SD依赖的软件为:

  • 驱动和固件
  • CANN开发套件包
  • CANN二进制算子包
  • PyTorch框架(支持版本为:2.1.0)
  • Torch_NPU
  • 推理引擎MindIE包

安装MindIE SD之前请提前安装以上依赖。详情请参见《MindIE安装指南》中的“安装MindIE > 方式二:物理机安装方式 >环境准备”章节

如果需要使用MindIE SD量化功能时,则必须安装CANN软件包NNAL神经网络加速库的Python库,即安装CANN软件包NNAL神经网络加速库时需添加--torch_atb参数进行指定,且运行时确保已安装Kernels算子包。安装命令如下所示:

./Ascend-cann-nnal_<version>_linux-<arch>.run --install --torch_atb

安装MindIE SD

MindIE SD不需要单独安装,安装MindIE时将会自动安装,MindIE的安装详情请参见《MindIE安装指南》中的“安装MindIE > 方式二:物理机安装方式 > 安装MindIE软件包”章节

配置环境变量

MindIE安装完成后,提供进程级环境变量设置脚本“set_env.sh”以自动完成环境变量设置,该环境变量设置脚本包含如表1所示中的ASCENDIE_HOME、LD_LIBRARY_PATH和ASCEND_CUSTOM_OPP_PATH,用户进程结束后自动失效。

表1 环境变量列表

环境变量

说明

ASCENDIE_HOME

MindIE RT软件包安装后文件存储路径。

LD_LIBRARY_PATH

动态库的查找路径。

ASCEND_CUSTOM_OPP_PATH

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

ASCEND_RT_VISIBLE_DEVICES

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

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