文档
注册

Ascend Extension for PyTorch

Ascend Extension for PyTorch插件是基于昇腾的深度学习适配框架,使昇腾NPU可以支持PyTorch框架,为PyTorch框架的使用者提供昇腾AI处理器的超强算力。

成长地图

由浅入深,带您玩转Ascend Extension for PyTorch

快速上手

快速了解Ascend Extension for PyTorch以及基于Ascend Extension for PyTorch进行模型迁移并训练的大致流程

模型开发

详细介绍了大模型的迁移及调优、精度问题定位、性能问题解决等指导,并提供了常见模型案例库

API 参考

介绍原生PyTorch框架的API接口支持情况与限制说明以及Ascend Extension for PyTorch自定义API的接口说明、功能描述、参数说明与使用示例

PyTorch2.2.0

PyTorch2.1.0

PyTorch1.11.0

配置与安装

详细介绍不同昇腾硬件产品形态的安装方案、环境搭建流程,以及卸载、其他常用操作等

安装与卸载

参考

套件与三方库

介绍了套件与三方库支持清单,提供了基于TorchAir的PyTorch图模式使用和MindSpeed迁移开发的指导

常见参考

提供了常见问题及故障处理的说明

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词