飞桨框架NPU版安装说明

安装方式:通过release/2.1分支源码编译安装

环境准备

昇腾NPU

  • 处理器: 鲲鹏920

  • 操作系统:Linux version 4.19.36-vhulk1907.1.0.h475.eulerosv2r8.aarch64

  • CANN: 20.3

  • Python版本: 2.7/3.6/3.7 (64 bit)

  • pip或pip3版本:9.0.1+ (64 bit)

  • cmake版本:3.15+

  • gcc/g++版本:8.2+

源码编译安装步骤:

第一步:下载Paddle源码并编译,CMAKE编译选项含义请参见编译选项表

# 下载源码,建议切换到 develop 分支
git clone -b develop https://github.com/PaddlePaddle/Paddle.git
cd Paddle

# 创建编译目录
mkdir build && cd build

# 执行cmake
cmake .. -DPY_VERSION=3 -DWITH_ASCEND=OFF -DPYTHON_EXECUTABLE=`which python3`  -DPYTHON_INCLUDE_DIR=`python3 -c "from distutils.sysconfig import get_python_inc; print(get_python_inc())"` -DWITH_ARM=ON -DWITH_ASCEND_CL=ON -DWITH_TESTING=ON -DCMAKE_BUILD_TYPE=Release -DON_INFER=ON -DWITH_XBYAK=OFF

# 使用以下命令来编译
make TARGET=ARMV8 -j$(nproc)

第二步:安装与验证编译生成的wheel包

编译完成之后进入Paddle/build/python/dist目录即可找到编译生成的.whl安装包,推荐设置PYTHONPATH环境变量安装:

# 设置PYTHONPATH
export PYTHONPATH=/your/dir/Paddle/build/python:$PYTHONPATH

# 验证命令
python -c "import paddle; paddle.utils.run_check()"