功能介绍

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

msFmkTransplt工具提供了0代码迁移工具和PyTorch GPU2Ascend脚本迁移工具两种方式,将基于GPU的脚本迁移为基于NPU的脚本的自动化方法,节省了人工手动进行脚本迁移的学习成本与工作量,大幅提升了迁移效率。同时提供PyTorch Analyse工具,帮助用户分析PyTorch训练脚本的算子、三方库API和动态shape的支持情况。

约束说明

环境准备

安装开发套件包,具体操作请参考CANN软件安装指南的“安装开发环境”章节。

配置环境变量

  1. 以运行用户登录,在任意目录下执行vi ~/.bashrc命令,打开.bashrc文件,在文件最后一行后面添加以下内容(以非root用户的默认安装路径为例)。
    # Ascend-cann-toolkit环境变量(请根据实际路径修改)
    source ~/Ascend/ascend-toolkit/set_env.sh
  2. 执行:wq!命令保存文件并退出。
  3. 执行source ~/.bashrc命令使其立即生效。