昇腾社区首页
中文
注册

7.0.0版本配套文档

文档名称

内容简介

更新说明

Ascend Extension for PyTorch 软件安装指南

提供在昇腾设备安装PyTorch框架训练环境,以及升级、卸载等操作。

新增支持PyTorch 2.5.1的适配,新增abi1版本安装torch_npu插件。

PyTorch 训练模型迁移调优指南

包含模型的迁移及调优、精度问题定位、性能问题解决等指导,并提供了常用模型案例库。

新增稠密模型调优案例和集群性能优化分析方法。

PyTorch 框架特性指南

基于Ascend Extension for PyTorch提供昇腾AI处理器的超强算力,从内存优化、报错定位、高性能计算等方面打造一系列独有特性。

第一次正式发布。

MindSpeed迁移开发指南

提供包括并行优化、内存优化、通信优化及计算优化以及关键场景等的特性使用指导。

新增适配Megatron-LM的core_r0.8.0分支,新增多模态能力相关特性说明。

PyTorch 图模式使用指南(TorchAir)

介绍将PyTorch的FX图转换为GE计算图,以及GE计算图的编译与执行接口。

新增图内标定SuperKernel范围和图内多流表达功能。

API参考

提供PyTorch2.5.1/2.4.0/2.3.1/2.1.0版本原生API接口在昇腾设备上的支持情况与限制说明,以及Ascend Extension for PyTorch自定义API的接口原型、功能描述、参数说明与调用示例等。

新增支持PyTorch 2.5.1的适配。

套件与三方库支持清单

介绍昇腾设备支持的模型套件与加速库、昇腾已原生支持的第三方库和昇腾自研插件。

新增端到端自定义算子开发和适配资料,提供端到端算子开发样例和说明。