昇腾社区首页
中文
注册
开发者
下载
支持

方式一：二进制软件包安装

通过wheel格式的二进制软件包直接安装PyTorch框架和torch_npu插件。

执行安装命令前，请参见安装前准备完成环境变量配置及其他环境准备。

安装PyTorch框架

表1 PyTorch框架安装

PyTorch版本

Python版本

系统架构

CANN版本

安装方式

安装命令

2.1.0

Python 3.8

AArch64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.1.0-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch-2.1.0-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

Python 3.8

X86_64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.1.0%2Bcpu-cp38-cp38-linux_x86_64.whl
# 安装命令
pip3 install torch-2.1.0+cpu-cp38-cp38-linux_x86_64.whl

Python 3.9

AArch64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.1.0-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch-2.1.0-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

Python 3.9

X86_64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.1.0%2Bcpu-cp39-cp39-linux_x86_64.whl
# 安装命令
pip3 install torch-2.1.0+cpu-cp39-cp39-linux_x86_64.whl

Python 3.10

AArch64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.1.0-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch-2.1.0-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

Python 3.10

X86_64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.1.0%2Bcpu-cp310-cp310-linux_x86_64.whl
# 安装命令
pip3 install torch-2.1.0+cpu-cp310-cp310-linux_x86_64.whl

Python 3.11

AArch64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.1.0-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch-2.1.0-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

Python 3.11

X86_64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.1.0%2Bcpu-cp311-cp311-linux_x86_64.whl
# 安装命令
pip3 install torch-2.1.0+cpu-cp311-cp311-linux_x86_64.whl

2.3.1

Python 3.8

AArch64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.3.1-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch-2.3.1-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

Python 3.8

X86_64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.3.1%2Bcpu-cp38-cp38-linux_x86_64.whl
# 安装命令
pip3 install torch-2.3.1+cpu-cp38-cp38-linux_x86_64.whl

Python 3.9

AArch64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.3.1-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch-2.3.1-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

Python 3.9

X86_64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.3.1%2Bcpu-cp39-cp39-linux_x86_64.whl
# 安装命令
pip3 install torch-2.3.1+cpu-cp39-cp39-linux_x86_64.whl

Python 3.10

AArch64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.3.1-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch-2.3.1-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

Python 3.10

X86_64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.3.1%2Bcpu-cp310-cp310-linux_x86_64.whl
# 安装命令
pip3 install torch-2.3.1+cpu-cp310-cp310-linux_x86_64.whl

Python 3.11

AArch64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.3.1-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch-2.3.1-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

Python 3.11

X86_64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.3.1%2Bcpu-cp311-cp311-linux_x86_64.whl
# 安装命令
pip3 install torch-2.3.1+cpu-cp311-cp311-linux_x86_64.whl

2.4.0

Python 3.8

AArch64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.4.0-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch-2.4.0-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

Python 3.8

X86_64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.4.0%2Bcpu-cp38-cp38-linux_x86_64.whl
# 安装命令
pip3 install torch-2.4.0+cpu-cp38-cp38-linux_x86_64.whl

Python 3.9

AArch64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.4.0-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch-2.4.0-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

Python 3.9

X86_64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.4.0%2Bcpu-cp39-cp39-linux_x86_64.whl
# 安装命令
pip3 install torch-2.4.0+cpu-cp39-cp39-linux_x86_64.whl

Python 3.10

AArch64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.4.0-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch-2.4.0-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

Python 3.10

X86_64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.4.0%2Bcpu-cp310-cp310-linux_x86_64.whl
# 安装命令
pip3 install torch-2.4.0+cpu-cp310-cp310-linux_x86_64.whl

Python 3.11

AArch64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.4.0-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch-2.4.0-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

Python 3.11

X86_64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.4.0%2Bcpu-cp311-cp311-linux_x86_64.whl
# 安装命令
pip3 install torch-2.4.0+cpu-cp311-cp311-linux_x86_64.whl

2.5.1

Python 3.9

AArch64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.5.1-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch-2.5.1-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

Python 3.9

X86_64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.5.1%2Bcpu-cp39-cp39-linux_x86_64.whl
# 安装命令
pip3 install torch-2.5.1+cpu-cp39-cp39-linux_x86_64.whl

Python 3.10

AArch64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.5.1-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch-2.5.1-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

Python 3.10

X86_64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.5.1%2Bcpu-cp310-cp310-linux_x86_64.whl
# 安装命令
pip3 install torch-2.5.1+cpu-cp310-cp310-linux_x86_64.whl

Python 3.11

AArch64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.5.1-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch-2.5.1-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

Python 3.11

X86_64

8.1.RC1

Pip

 
# 下载软件包
wget https://download.pytorch.org/whl/cpu/torch-2.5.1%2Bcpu-cp311-cp311-linux_x86_64.whl
# 安装命令
pip3 install torch-2.5.1+cpu-cp311-cp311-linux_x86_64.whl

更多PyTorch版本可单击Link进行查询。

安装torch_npu插件

表2 安装torch_npu插件

PyTorch版本

torch_npu插件版本

Python版本

系统架构

CANN版本

安装方式

安装命令

2.1.0

7.0.0

Python 3.8

AArch64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.1.0/torch_npu-2.1.0.post12-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch_npu-2.1.0.post12-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

7.0.0

Python 3.8

X86_64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.1.0/torch_npu-2.1.0.post12-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
# 安装命令
pip3 install torch_npu-2.1.0.post12-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

7.0.0

Python 3.9

AArch64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.1.0/torch_npu-2.1.0.post12-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch_npu-2.1.0.post12-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

7.0.0

Python 3.9

X86_64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.1.0/torch_npu-2.1.0.post12-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
# 安装命令
pip3 install torch_npu-2.1.0.post12-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

7.0.0

Python 3.10

AArch64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.1.0/torch_npu-2.1.0.post12-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch_npu-2.1.0.post12-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

7.0.0

Python 3.10

X86_64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.1.0/torch_npu-2.1.0.post12-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
# 安装命令
pip3 install torch_npu-2.1.0.post12-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

7.0.0

Python 3.11

AArch64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.1.0/torch_npu-2.1.0.post12-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch_npu-2.1.0.post12-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

7.0.0

Python 3.11

X86_64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.1.0/torch_npu-2.1.0.post12-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
# 安装命令
pip3 install torch_npu-2.1.0.post12-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

2.3.1

7.0.0

Python 3.8

AArch64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.3.1/torch_npu-2.3.1.post6-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch_npu-2.3.1.post6-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

7.0.0

Python 3.8

X86_64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.3.1/torch_npu-2.3.1.post6-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
# 安装命令
pip3 install torch_npu-2.3.1.post6-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

7.0.0

Python 3.9

AArch64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.3.1/torch_npu-2.3.1.post6-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch_npu-2.3.1.post6-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

7.0.0

Python 3.9

X86_64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.3.1/torch_npu-2.3.1.post6-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
# 安装命令
pip3 install torch_npu-2.3.1.post6-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

7.0.0

Python 3.10

AArch64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.3.1/torch_npu-2.3.1.post6-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch_npu-2.3.1.post6-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

7.0.0

Python 3.10

X86_64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.3.1/torch_npu-2.3.1.post6-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
# 安装命令
pip3 install torch_npu-2.3.1.post6-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

7.0.0

Python 3.11

AArch64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.3.1/torch_npu-2.3.1.post6-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch_npu-2.3.1.post6-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

7.0.0

Python 3.11

X86_64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.3.1/torch_npu-2.3.1.post6-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
# 安装命令
pip3 install torch_npu-2.3.1.post6-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

2.4.0

7.0.0

Python 3.8

AArch64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.4.0/torch_npu-2.4.0.post4-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch_npu-2.4.0.post4-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

7.0.0

Python 3.8

X86_64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.4.0/torch_npu-2.4.0.post4-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
# 安装命令
pip3 install torch_npu-2.4.0.post4-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

7.0.0

Python 3.9

AArch64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.4.0/torch_npu-2.4.0.post4-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch_npu-2.4.0.post4-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

7.0.0

Python 3.9

X86_64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.4.0/torch_npu-2.4.0.post4-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
# 安装命令
pip3 install torch_npu-2.4.0.post4-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

7.0.0

Python 3.10

AArch64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.4.0/torch_npu-2.4.0.post4-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch_npu-2.4.0.post4-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

7.0.0

Python 3.10

X86_64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.4.0/torch_npu-2.4.0.post4-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
# 安装命令
pip3 install torch_npu-2.4.0.post4-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

7.0.0

Python 3.11

AArch64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.4.0/torch_npu-2.4.0.post4-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch_npu-2.4.0.post4-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

7.0.0

Python 3.11

X86_64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.4.0/torch_npu-2.4.0.post4-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
# 安装命令
pip3 install torch_npu-2.4.0.post4-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

2.5.1

7.0.0

Python 3.9

AArch64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.5.1/torch_npu-2.5.1-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch_npu-2.5.1-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

7.0.0

Python 3.9

X86_64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.5.1/torch_npu-2.5.1-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
# 安装命令
pip3 install torch_npu-2.5.1-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

7.0.0

Python 3.10

AArch64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.5.1/torch_npu-2.5.1-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch_npu-2.5.1-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

7.0.0

Python 3.10

X86_64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.5.1/torch_npu-2.5.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
# 安装命令
pip3 install torch_npu-2.5.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

7.0.0

Python 3.11

AArch64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.5.1/torch_npu-2.5.1-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
# 安装命令
pip3 install torch_npu-2.5.1-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

7.0.0

Python 3.11

X86_64

8.1.RC1

Pip

 
# 下载插件包
wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.5.1/torch_npu-2.5.1-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
# 安装命令
pip3 install torch_npu-2.5.1-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

更多PyTorch插件版本可单击Link查询。

若PyTorch版本为2.1.0，出现“找不到google或者protobuf或者protobuf版本过高”报错时，需执行如下命令：
pip3 install protobuf==3.20

验证是否成功安装

  • 方法一
    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')
父主题： 安装PyTorch