文档
注册

概述

昇腾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优化器不支持使用工具进行迁移,若原始代码中包含该优化器,用户需自行修改。
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词