入门概述

简介

Ascend Extension for PyTorch插件实现了PyTorch框架与昇腾AI处理器的适配，用户只需对原有代码进行最小改动，便可完成模型从其他硬件平台向昇腾平台的迁移。此外，该插件支持自动微分、动态分发、调试等功能，并且易于扩展，能够快速适配新的网络类型。

本文档以CNN模型为例介绍如何进行迁移和训练PyTorch模型。

使用说明

该产品支持情况如下表。

使用流程

在昇腾AI处理器上搭建开发&运行环境并进行PyTorch模型迁移训练，整体流程如下图所示：

图1 PyTorch模型迁移训练整体流程

环境准备

• 准备Atlas A2 训练系列产品的服务器，并安装对应的驱动和固件，具体安装过程请参见《CANN 软件安装指南》中的“安装NPU驱动和固件”章节。
• 安装Toolkit开发套件包，具体安装过程请参见《CANN 软件安装指南》的“准备安装用户和运行用户”以及“安装CANN（物理机场景）”章节。
• 安装PyTorch框架及torch_npu插件，具体安装过程请参见《Ascend Extension for PyTorch 软件安装指南》。