昇腾社区首页
中文
注册

Ascend Extension for PyTorch

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

  • 版本说明

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

  • 产品概述

    介绍Ascend Extension for PyTorch的整体架构和关键功能。

  • 快速体验

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

环境准备

大模型训练加速库

已适配传统模型训练

  • 支持模型列表

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

未适配传统模型迁移

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

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

参考
  • 环境变量

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

  • 常见问题

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

  • 故障处理

    训练和推理过程中异常故障的处理流程以及故障案例。