飞桨框架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编译选项含义请参见编译选项表

  1. # 下载源码,建议切换到 develop 分支
  2. git clone -b develop https://github.com/PaddlePaddle/Paddle.git
  3. cd Paddle
  4. # 创建编译目录
  5. mkdir build && cd build
  6. # 执行cmake
  7. 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
  8. # 使用以下命令来编译
  9. make TARGET=ARMV8 -j$(nproc)

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

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

  1. # 设置PYTHONPATH
  2. export PYTHONPATH=/your/dir/Paddle/build/python:$PYTHONPATH
  3. # 验证命令
  4. python -c "import paddle; paddle.utils.run_check()"