安装指南

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

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

准备安装 PyTorch：

PyTorch版本
PyTorch-NPU版本
CANN-toolkit版本
CPU架构
安装方式
PyTorch版本
2.3.1
2.2.0
2.1.0
PyTorch-NPU版本
null
CANN-toolkit版本
null
CPU架构
x86-64
aarch64
安装方式
Docker
pip
源码构建

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的值即为安装成功。