昇腾社区首页
中文
注册

安装说明

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

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

安装方案

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

图1 安装方案

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

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

安装方式

本手册提供了多种安装方式,用户可根据实际需求自行选择安装PyTorch框架和torch_npu插件的方式。