使用迁移工具
- 进入迁移工具所在路径。
cd Ascend-cann-toolkit安装目录/ascend-toolkit/latest/tools/ms_fmk_transplt/
- 执行脚本迁移任务,参考表1配置信息。
./pytorch_gpu2npu.sh -i 原始脚本路径 -o 脚本迁移结果输出路径 -v 原始脚本的PyTorch框架版本
命令示例:
./pytorch_gpu2npu.sh -i /home/user -o /home/out -v 2.1.0
- 切换目录至迁移完成后的训练脚本所在路径(以/home/user为例),执行以下命令使用虚拟数据集进行训练,迁移完成后的训练脚本可在NPU上正常运行。开始打印迭代日志,说明训练功能迁移成功。
cd /home/user python main.py -a resnet50 --gpu 1 --epochs 1 --dummy # --gpu 1表示使用卡1,--epochs 1是指迭代次数为1
- 完成脚本迁移,进入脚本迁移结果的输出路径查看结果件。
父主题: 迁移训练模型