编译安装指南¶
编译说明¶
本文档介绍如何从源码编译MindIE-LLM,生成 .whl 包,安装与运行。
环境准备¶
镜像安装方式¶
MindIE镜像获取请参见镜像安装方式。
容器/物理机安装方式¶
编译安装¶
-
安装Python工具。MindIE-LLM 支持 Python == 3.10 和 Python == 3.11。
-
克隆源码仓库。
-
编译第三方依赖。
-
设置环境变量。 获取 Python site-packages 路径(建议不要硬编码 torch 路径),并配置动态库搜索路径:
TORCH_PATH=$(python3 -c "import torch, os; print(os.path.dirname(torch.__file__))") TORCH_NPU_PATH=$(python3 -c "import torch_npu, os; print(os.path.dirname(torch_npu.__file__))") export LD_LIBRARY_PATH=${TORCH_PATH}/lib:${TORCH_PATH}/../torch.libs:$LD_LIBRARY_PATH export PYTORCH_NPU_INSTALL_PATH=${TORCH_NPU_PATH}可选:指定生成
.whl包的版本号: -
编译生成 MindIE-LLM 的
.whl包。 在源码根目录下执行:- 编译完成后,会在当前目录生成
mindie_llm-<version>-*.whl文件。 - 编译时,
setup.py会自动调用build.sh编译C++代码,并拷贝第三方依赖到包内。 - 编译后,生成临时目录
build、存放二进制的目录output和 debug 符号表llm_debug_symbols目录。
- 编译完成后,会在当前目录生成
-
安装MindIE-LLM。
-
编译ATB_Models的
.whl包。注意:使用 Python 3.10 环境编译,需配套 torch 2.9.0 版本 + torch_npu 2.9.0 版本, 否则会导致 _bz2 模块缺失,从而导致编译失败。
-
安装 ATB_Models。
-
配置运行环境变量(可选)。
ATB_Models 安装完成后,需设置其运行所需的环境变量。通过 Python 动态获取
atb_llm的安装路径,以适配不同的 Python 环境和site-packages位置:ATB_LLM_PATH=$(python3 -c "import atb_llm, os; print(os.path.dirname(atb_llm.__file__))") export ATB_SPEED_HOME_PATH=${ATB_LLM_PATH} export LD_LIBRARY_PATH=${ATB_LLM_PATH}/lib:${LD_LIBRARY_PATH}提示
- 建议将以上命令写入
~/.bashrc或启动脚本中,避免每次手动设置。 - MindIE 镜像内置该环境变量,如使用镜像则无需手动设置。
- 建议将以上命令写入
升级¶
升级详情请参见升级。
卸载¶
卸载详情请参见卸载。