方式三：二进制软件包安装（abi1版本）

安装PyTorch框架

# 下载软件包
wget https://download.pytorch.org/whl/cpu-cxx11-abi/torch-<version>%2Bcpu.cxx11.abi-cp3x-cp3x-linux_x86_64.whl

# 安装命令
pip3 install torch-<version>+cpu.cxx11.abi-cp3x-cp3x-linux_x86_64.whl

示例命令如下：

wget https://download.pytorch.org/whl/cpu-cxx11-abi/torch-2.1.0%2Bcpu.cxx11.abi-cp39-cp39-linux_x86_64.whl
pip3 install torch-2.1.0+cpu.cxx11.abi-cp39-cp39-linux_x86_64.whl

• 执行安装命令前，请参见安装前准备完成环境变量配置及相关依赖的安装。
• x表示8、9、10或11，即当前支持Python 3.8、Python 3.9、Python 3.10和Python 3.11。
• <version>表示PyTorch框架版本，当前指2.1.0、2.3.1、2.4.0或2.5.1。
• 当前暂不支持AArch64架构的abi1版本安装。

安装torch_npu插件

如下操作以PyTorch 2.1.0版本为例，介绍如何获取abi1版本的安装包并安装torch_npu插件。

1. 下载abi1版本安装包。

   wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.1.0/torch_npu-2.1.0.post12_cxx11.abi_x86_64.zip

   以v2.1.0-7.0.0为例，下载对应的Ascend Extension for PyTorch安装包，其他分支请参见《版本说明》中的“相关产品版本配套说明”章节。

2. 解压缩安装包。

   unzip -o torch_npu-2.1.0.post12_cxx11.abi_x86_64.zip

3. 安装abi1版本torch_npu插件。

   # 选择对应Python版本的安装包，如Python 3.9
   pip3 install torch_npu-2.1.0.post12+cxx11.abi-cp39-cp39-manylinux_2_28_x86_64.whl

• 执行安装命令前，请参见安装前准备完成环境变量配置及相关依赖的安装。
• PyTorch框架版本为2.1.0、2.3.1或者2.4.0，Python版本支持3.8~3.11；PyTorch框架版本为2.5.1，Python版本支持3.9~3.11。
• 当前暂不支持AArch64架构的abi1版本安装。

验证是否成功安装

• 方法一

  python3 -c "import torch;import torch_npu; a = torch.randn(3, 4).npu(); print(a + a);"

  输出如下类似信息说明PyTorch框架与插件安装成功。

  1 2 3

  tensor([[-0.6066, 6.3385, 0.0379, 3.3356], [ 2.9243, 3.3134, -1.5465, 0.1916], [-2.1807, 0.2008, -1.1431, 2.1523]], device='npu:0')

• 方法二

  1 2 3 4 5 6 7 8

  import torch import torch_npu x = torch.randn(2, 2).npu() y = torch.randn(2, 2).npu() z = x.mm(y) print(z)

  输出如下类似信息说明PyTorch框架与插件安装成功。

  1 2	tensor([[-0.0515, 0.3664], [-0.1258, -0.5425]], device='npu:0')