Atlas系列硬件: Atlas 200l DK A2 开发者套件; Atlas 200l DK 开发者套件（型号：3000）; 加速模块、加速卡、智能小站、服务器、集群

异构计算架构: CANN社区版; CANN商用版

AI框架: MindSpore; TensorFlow框架模型（社区版）; TensorFlow框架模型（商用版）; PyTorch框架适配

昇腾推理引擎: MindIE

应用使能MindX: MindX DL; MindX Edge; MindX SDK

全流程开发工具: MindStudio

安装前必读

为使用PyTorch框架的开发者提供昇腾AI处理器的超强算力，昇腾开发PyTorch Ascend Adapter插件（torch_npu）用于适配PyTorch框架，本章节指导用户在昇腾环境安装PyTorch框架和torch_npu插件。

若用户仅进行离线推理，请跳过此章节。

前提条件

安装配套版本的CANN软件并配置CANN环境变量。
安装二进制算子包具体请参考安装二进制算子包。
安装对应框架版本的torchvision，PyTorch 1.11.0需安装0.12.0版本，PyTorch 2.0.1版本需安装0.15.2版本，PyTorch 2.1.0版本需安装0.16.0版本。
执行如下命令安装torchvision，以0.12.0版本为例：
```
pip3 install torchvision==0.12.0 
```
通过源码编译安装torch_npu插件时，安装如下环境依赖。
执行如下命令安装。如果使用非root用户安装，需要在命令后加--user，例如：pip3 install pyyaml --user。
```
pip3 install pyyaml
pip3 install wheel
pip3 install setuptools
```

版本配套关系

表1 PyTorch版本配套关系
CANN版本	PyTorch版本	PyTorch Ascend Adapter插件版本	PyTorch Ascend Adapter插件代码分支名称	PyTorch Ascend Adapter插件安装包版本	Python版本	AscendHub镜像版本/名称
7.0.RC1	PyTorch1.11.0	5.0.RC3	v1.11.0-5.0.rc3	1.11.0.post4	Python3.7.x（3.7.5及以上）、Python3.8.x、Python3.9.x、Python3.10.x	-
	PyTorch2.0.1		v2.0.1-5.0.rc3	2.0.1	Python3.8.x、Python3.9.x、Python3.10.x	-
	PyTorch2.1.0		v2.1.0-5.0.rc3	2.1.0.rc1	Python3.8.x、Python3.9.x、Python3.10.x	-

安装步骤

在昇腾环境安装PyTorch步骤如下：

安装PyTorch框架。
安装torch_npu插件：可以直接下载whl包快速安装也可以通过源码编译安装。
（可选）编译libtorch_npu：需要进行libtorch推理测试时执行。
（可选）安装APEX模块：混合精度训练时安装。

若用户希望在容器中安装PyTorch，可以根据表1，前往AscendHub昇腾PyTorch镜像仓库选择下载并安装对应版本的镜像，之后在容器中使用上述步骤安装PyTorch。
如果用户不确定自己的软件包版本，可参考查询软件包版本信息进行查询。

父主题： 安装PyTorch