昇腾社区首页
中文
注册

Ascend Extension for PyTorch

帮助开发者认识、了解昇腾Ascend Extension for PyTorch,以及如何基于此进行模型开发和调优。

  • 版本说明

    介绍Ascend Extension for PyTorch软件版本信息。

  • 快速入门

    以CNN模型为例,介绍将其迁移至NPU上运行的方法。

环境准备

大模型训练加速库

  • MindSpeed迁移开发

    大模型训练加速库迁移,昇腾MindSpeed训练加速库,提供丰富算法和模型,使能训练全流程加速。

已适配传统模型训练

  • 支持模型列表

    已适配的模型清单,可参考对应模型的readme进行训练和推理。

未适配传统模型迁移

模型迁移与调优
三方库列表
API
  • API参考

    原生PyTorch API在昇腾设备上的支持情况,及自定义API的使用说明。

参考
  • 环境变量

    训练和推理过程中可使用的环境变量。

  • 故障处理

    训练和推理过程中异常故障的处理流程以及故障分析指导。

  • 常见问题

    在迁移、调优等过程中常见问题的处理方法。

  • 安全声明

    Ascend Extension for PyTorch及相关组件的安全加固措施建议。