安装PyTorch

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

为使用PyTorch框架的开发者提供昇腾AI处理器的超强算力，昇腾开发Ascend Extension for PyTorch插件用于适配PyTorch框架，请参考《Ascend Extension for PyTorch 配置与安装》的如下章节在昇腾环境安装PyTorch框架和Ascend Extension for PyTorch插件。

1. 安装前必读
2. 安装PyTorch框架
3. 安装torch_npu插件。
   参考"源码编译安装torch_npu插件"章节进行安装，此过程中需要将分支代码修改成如下对应版本。

   表1 分支代码获取方式

   版本号	获取方式
   2.1.0	git clone -b v6.0.0.alpha001-pytorch2.1.0 https://gitee.com/ascend/pytorch.git
   2.3.1	git clone -b v6.0.0.alpha001-pytorch2.3.1 https://gitee.com/ascend/pytorch.git
   2.4.0	git clone -b v6.0.0.alpha001-pytorch2.4.0 https://gitee.com/ascend/pytorch.git

4. 安装APEX模块。