昇腾社区首页
中文
注册
开发者
下载

安装说明

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

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

安装方案

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

图 1 安装方案[object Object][object Object]

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

表 1 产品硬件支持列表

[object Object]undefined
[object Object]
  • 各硬件产品对应物理机部署场景支持的操作系统请参考
  • 各硬件产品对应虚拟机部署场景支持的操作系统请参考《CANN 软件安装指南》的“”章节(商用版)或“”章节(社区版)。
  • 各硬件产品对应容器部署场景支持的操作系统请参考《CANN 软件安装指南》的“”章节(商用版)或“”章节(社区版)。

安装方式

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