使用迁移工具

进入迁移工具所在路径。

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
```
完成脚本迁移，进入脚本迁移结果的输出路径查看结果件。

脚本迁移过程中会启动迁移分析，默认使用torch_apis和affinity_apis的分析模式，可参见查看分析报告查看对应的结果件。
成功保存权重，说明保存权重功能迁移成功。

父主题： 迁移训练模型