（推荐）自动迁移

简介

仅PyTorch 1.8.1版本及以上使用，自动迁移方式较简单，且修改内容最少，只需在训练脚本中导入库代码。

使用方法

1. 配置环境变量。

   export PYTHONPATH={CANN包安装目录}/ascend-toolkit/latest/tools/ms_fmk_transplt/torch_npu_bridge:$PYTHONPATH

2. 在训练脚本中导入以下库代码。

   import torch
   import torch_npu
   .....
   import transfer_to_npu

3. 参考模型训练执行训练。查看训练后是否生成权重文件，生成了如下图文件则说明迁移训练成功。