本文档主要针对昇腾 NPU、寒武纪 MLU、昆仑 XPU 硬件平台,介绍 PaddleX 使用指南。
根据所属硬件平台拉取镜像、安装飞桨,请参考多硬件飞桨安装
使用下述命令从 GitHub 获取 PaddleX 最新源码。
git clone https://github.com/PaddlePaddle/PaddleX.git
如果访问 GitHub 网速较慢,可以从 Gitee 下载,命令如下:
git clone https://gitee.com/paddlepaddle/PaddleX.git
参考下述命令,按提示操作,完成 PaddleX 依赖的安装。
cd PaddleX
# 注意,如果是 arm64 架构的机器,需要先手动安装 tool_helpers,x86 架构机器不需要
# python3.9 版本,适用 npu arm64 架构机器
pip install https://paddle-model-ecology.bj.bcebos.com/paddlex/whl/paddlenlp-device/tool_helpers-0.1.1-cp39-cp39-linux_aarch64.whl
# python3.10 版本,适用 xpu arm64 架构机器
pip install https://paddle-model-ecology.bj.bcebos.com/paddlex/whl/paddlenlp-device/tool_helpers-0.1.1-cp310-cp310-linux_aarch64.whl
# 安装 PaddleX whl
# -e:以可编辑模式安装,当前项目的代码更改,都会直接作用到已经安装的 PaddleX Wheel
pip install -e .
# 安装 PaddleX 相关依赖
paddlex --install
# 完成安装后会有如下提示:
# All packages are installed.
PaddleX 模型训练/评估/推理的详细使用方法,参考文档:PaddleX 单模型开发工具 在具体设备上,根据所属硬件平台,添加配置设备的参数,即可在对应硬件上使用上述工具。