[Atlas200I A2] 安装torch-npu

在Atlas200I A2 DK上安装torch-npu碰了壁，一开始还以为该设备不支持安装torch-npu。后面浏览昇腾论坛帖子atlas 200i dk a2的310b4芯片是否支持torch-npu、BrokenPipeError_Atlas 200I DK A2_昇腾论坛，依照着里面的信息自己尝试了一下，成功安装上了torch-npu。现将步骤记录如下：

1.安装cann-toolkit：Ascend-cann-toolkit_7.0.0_linux-aarch64.run

2.使用conda创建python环境：

conda create -n test python=3.9

3.构建toolkit环境变量：

source /usr/local/Ascend/ascend-toolkit/set_env.sh

4.在合适的路径下wget获取torch-npu安装包：

wget https://gitee.com/ascend/pytorch/releases/download/v6.0.rc1-pytorch2.1.0/torch_npu-2.1.0.post3-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

5.pip安装whl文件：

pip install /home/HwHiAiUser/Downloads/torch_npu-2.1.0.post3-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

6.pip安装numpy：

pip install "numpy<2.0"

7.输入测试代码看看环境配好没有：

python3 -c "import torch;import torch_npu;print(torch_npu.npu.is_available())"

8.可选，使用下列代码解决pip check出现的问题：

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple  decorator attrs psutil absl-py cloudpickle scipy tornado synr==0.5.0

安装成功了的话，第7步会输出True。如下图所示。