昇腾社区首页
中文
注册

方式三:二进制软件包安装(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插件。

  1. 下载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安装包,其他分支请参见《版本说明》中的“相关产品版本配套说明”章节

  2. 解压缩安装包。
    unzip -o torch_npu-2.1.0.post13_cxx11.abi_x86_64.zip
  3. 安装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版本安装。