安装指南 — 昇腾开源 文档 (original) (raw)

安装指南

跟随指导，安装在NPU上运行的PyTorch版本。

1. 选择需要安装的 PyTorch 版本

准备安装 PyTorch：

PyTorch版本

PyTorch-NPU版本

CANN-toolkit版本

CPU架构

安装方式

PyTorch版本

2.4.0

2.3.1

2.2.0

2. 安装 PyTorch

警告

如果使用了非CANN安装时的Python环境（如Conda），请确保CANN-toolkit依赖的Python包在该环境中已经 安装 。

备注

请确保已经安装了与上述CANN-toolkit版本匹配的驱动和固件。

备注

请确保已经根据上述表格建议安装了对应的CANN-toolkit版本以及相应的固件和驱动，并应用了CANN-toolkit环境变量。

备注

请确保已经根据上述表格建议安装了对应的CANN-toolkit版本以及相应的固件和驱动，并应用了CANN-toolkit环境变量。

2.1 环境依赖

• Python 3.8 ~ 3.10
• 支持C++17的编译器，例如clang 或者 gcc (9.4.0及以上)
• Conda

备注

请确认CXX11_ABI是关闭的，如果无法确定，建议显式关闭：

export _GLIBCXX_USE_CXX11_ABI=0

2.2 构建

3. 验证安装结果

1import torch 2import torch_npu 3 4x = torch.randn(2, 2).npu() 5y = torch.randn(2, 2).npu() 6z = x.mm(y) 7 8print(z)

程序能够成功打印矩阵Z的值即为安装成功。