飞桨框架昇腾NPU版训练示例

Mnist运行示例：

下载模型和运行脚本

  1. # 下载模型和数据
  2. wget https://fleet.bj.bcebos.com/ascend/npu.tar.gz
  3. tar xzvf npu.tar.gz

执行训练

  1. sh run_mnist.sh

配置说明

run_mnist.sh中的配置

  1. # 配置paddle的路径
  2. export PYTHONPATH=/your/dir/Paddle/build/python:$PYTHONPATH
  3. # 配置使用的NPU卡
  4. export FLAGS_selected_npus=5
  5. # 执行训练脚本
  6. python3 mnist.py