PyTorch模型开发
帮助开发者认识、了解昇腾模型,以及如何基于昇腾Ascend Extension for PyTorch进行模型开发。
文档总览
文档 |
内容介绍 |
关键资源获取 |
---|---|---|
产品概述 |
快速了解Ascend Extension for PyTorch的架构及关键功能特性。 |
|
配置与安装 |
详细介绍不同昇腾硬件产品形态的安装方案、环境搭建流程,以及卸载、其他常用操作等。 |
|
快速体验 |
以一个简单的模型迁移样例,采用了最简单的自动迁移方法,帮助用户快速体验其他平台的模型脚本迁移到昇腾NPU上的流程,并通过修改脚本代码,使其可以迁移到昇腾NPU上进行训练。 |
|
模型开发 |
包含了大模型的迁移及调优、精度问题定位、性能问题解决等指导,并提供了常见模型案例库。 |
|
套件与三方库 |
包含了套件与三方库支持清单,提供了基于TorchAir的PyTorch图模式使用和MindSpeed迁移开发的指导。 |
|
API参考 |
主要介绍原生PyTorch框架2.4.0/2.3.1/2.1.0的API接口支持情况与限制说明以及Ascend Extension for PyTorch自定义API的接口说明、功能描述、参数说明与使用示例。 |
|
环境变量 |
包含了开发者基于CANN及Ascend Extension for PyTorch构建AI应用和训练执行中可使用的环境变量。 |