在华为npu上安装存在各种版本依赖，而它们又不是在一个文档中，并且华为文档一直在更新，因此记录下来，免得每次要安装新环境都折磨自己。

  1. 安装驱动和固件

    已安装，版本：8.1.RC1

  2. 安装CANN

    第一步，安装依赖

    1
    pip3 install attrs cython numpy==1.24.0 decorator sympy cffi pyyaml pathlib2 psutil protobuf==3.20 scipy requests absl-py --user

    第二部，安装软件及相应算子

  3. 安装Pytorch

    以pytorch 2.1.0为例：

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    CANN版本：8.1.RC1
    python版本：3.10
    # 下载软件包 torch
    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


    # 下载插件包 torch_npu
    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

    4. 验证是否安装成功

    1
    python3 -c "import torch;import torch_npu; a = torch.randn(3, 4).npu(); print(a + a);"