方式三:二进制软件包安装(abi1版本)
本章节介绍如何获取并安装abi1版本的PyTorch框架和torch_npu插件。
安装PyTorch框架
使用以下命令可以直接从PyTorch官方获取abi1版本的安装包。
# 下载软件包 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.5.1或2.6.0。
- 当前暂不支持AArch64架构的abi1版本安装。
安装torch_npu插件
如下操作以PyTorch 2.1.0版本为例,介绍如何获取abi1版本的安装包并安装torch_npu插件。
- 下载abi1版本安装包。
wget https://gitee.com/ascend/pytorch/releases/download/v7.1.0-pytorch2.1.0/torch_npu-2.1.0.post13_cxx11.abi_x86_64.zip
以v2.1.0-7.1.0为例,下载对应的Ascend Extension for PyTorch安装包,其他分支请参见《版本说明》中的“相关产品版本配套说明”章节。
- 解压缩安装包。
unzip -o torch_npu-2.1.0.post13_cxx11.abi_x86_64.zip
- 安装abi1版本torch_npu插件。
# 选择对应Python版本的安装包,如Python 3.9 pip3 install torch_npu-2.1.0.post13+cxx11.abi-cp39-cp39-manylinux_2_28_x86_64.whl

- 执行安装命令前,请参见安装前准备完成环境变量配置及相关依赖的安装。
- PyTorch框架版本为2.1.0,Python版本支持3.8~3.11;PyTorch框架版本为2.5.1或2.6.0,Python版本支持3.9~3.11。
- 当前暂不支持AArch64架构的abi1版本安装。
父主题: 安装PyTorch