昇腾社区首页
中文
注册

安装说明

为使用PyTorch框架的开发者提供昇腾AI处理器的超强算力,昇腾开发Ascend Extension for PyTorch(即torch_npu插件)用于适配PyTorch框架。

本文主要向用户介绍如何快速完成PyTorch框架、Ascend Extension for PyTorch(即torch_npu插件)以及扩展模块的安装。

安装方案

本文档包含物理机、容器、虚拟机场景下,安装驱动、固件、CANN软件、PyTorch框架和torch_npu插件的方案,部署架构如图1所示。

图1 安装方案

硬件配套和支持的操作系统

表1 产品硬件支持列表

产品

是否支持(训练场景)

Atlas A3 训练系列产品

Atlas A3 推理系列产品

x

Atlas A2 训练系列产品

Atlas 800I A2 推理产品

x

Atlas 200I/500 A2 推理产品

x

Atlas 推理系列产品

x

Atlas 训练系列产品

  • 各硬件产品对应物理机部署场景支持的操作系统请参考兼容性查询助手
  • 各硬件产品对应虚拟机部署场景支持的操作系统请参考《CANN软件安装指南》的“安装说明”章节
  • 各硬件产品对应容器部署场景支持的操作系统请参考《CANN软件安装指南》的“安装说明”章节

安装方式

本手册提供了二进制软件包、源码编译以及二进制软件包(abi1版本)安装方式,用户可根据实际需求自行选择安装PyTorch框架和torch_npu插件的方式,不要求两者安装方式统一。