Ascend Extension for PyTorch
帮助开发者认识、了解昇腾模型,以及如何基于昇腾Ascend Extension for PyTorch进行模型开发。
- 版本说明
介绍Ascend Extension for PyTorch软件版本信息。
- 产品概述
介绍Ascend Extension for PyTorch的整体架构和关键功能。
- 快速体验
以CNN模型为例,介绍将其迁移至NPU上运行的方法。
环境准备
- 配置与安装
在昇腾设备上搭建PyTorch训练环境。
大模型训练加速库
- 大模型训练加速库迁移
昇腾MindSpeed训练加速库,提供丰富算法和模型,使能训练全流程加速。
已适配传统模型训练
- 支持模型列表
提供已适配的模型清单,可参考对应模型的readme进行训练和推理。