概述

昇腾NPU是AI算力的后起之秀，但目前训练和在线推理脚本大多是基于GPU的。由于NPU与GPU的架构差异，基于GPU的训练和在线推理脚本不能直接在NPU上使用。

PyTorch GPU2Ascend工具提供了将基于GPU的脚本迁移为基于NPU的脚本的自动化方法，节省了人工手动进行脚本迁移的学习成本与工作量，大幅提升了迁移效率。同时提供分析方法，帮助用户分析PyTorch训练脚本的算子支持情况。当前支持1.11.0、2.0.1、2.1.0版本的训练脚本分析和迁移。

约束说明

PyTorch GPU2Ascend工具迁移后脚本的执行逻辑与迁移前保持一致。
待迁移脚本需要在GPU环境下且基于Python3.7及以上能够跑通。
若原始代码中调用了三方库，迁移过程可能会存在适配问题。在迁移原始代码前，用户需要根据已调用的三方库，自行安装昇腾已适配的三方库版本，已适配的三方库信息和使用指南请参考《PyTorch网络迁移和训练指南》中“模型套件和第三方库”章节。
APEX中使用的FusedAdam优化器不支持使用工具进行迁移，若原始代码中包含该优化器，用户需自行修改。

父主题： PyTorch GPU2Ascend