1.【新教程】华为昇腾NPU的pytorch环境搭建 - Lukea - 博客园

1、新建conda环境。

conda create -n pytorch python=3.10

2、在新建好的conda环境中，安装基础的依赖。

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

3.安装Pytoch：下面是对应pytorch版本链接，注意架构是x86还是aarch64，然后pip install

wget https://download.pytorch.org/whl/cpu/torch-2.1.0-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl#sha256=de7d63c6ecece118684415a3dbd4805af4a4c1ee1490cccf7405d8c240a481b4

pytorch: Ascend Extension for PyTorch

4.安装pytorch-npu，找到自己适配的版本，然后pip install

https://gitee.com/ascend/pytorch/releases/tag/v6.0.0-pytorch2.1.0

社区版资源下载-资源下载中心-昇腾社区
下载好指定版本后，使用pip install 安装下载好的包即可

5、测试。
使用以下命令测试，如果正常输出结果，说明环境搭建成功。

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

华为昇腾服务器（固件版本查询、驱动版本查询、CANN版本查询）

华为昇腾服务器（固件版本查询、驱动版本查询、CANN版本查询）_怎么查看cann版本-CSDN博客