在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。如下图所示。

