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
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);"