安装指南 =========================== 跟随指导，安装在NPU上运行的PyTorch版本。 1. 选择需要安装的 PyTorch 版本 ------------------------------ 准备安装 PyTorch： .. raw:: html
PyTorch版本
PyTorch-NPU版本
CANN-toolkit版本
CPU架构
安装方式
PyTorch版本
2.4.0
2.3.1
2.2.0
PyTorch-NPU版本
null
CANN-toolkit版本
null
CPU架构
x86-64
aarch64
安装方式
Docker
pip
源码构建
2. 安装 PyTorch ---------------- .. warning:: 如果使用了非CANN安装时的Python环境（如Conda），请确保CANN-toolkit依赖的Python包在该环境中已经 `安装 <../ascend/quick_install.html>`_ 。 .. raw:: html

备注

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


        

    

    

        

                
备注

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

        

        

            

        

    

    

        

            

                
备注

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

            

            
2.1 环境依赖

            
        

        

            

                
备注

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

            

            
export _GLIBCXX_USE_CXX11_ABI=0

            
2.2 构建

                

                    

                

        

    



3. 验证安装结果
------------------

.. code-block:: python
    :linenos:

    import torch
    import torch_npu

    x = torch.randn(2, 2).npu()
    y = torch.randn(2, 2).npu()
    z = x.mm(y)

    print(z)

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