尝试了几十次最终才差不多,主要总结几个可能不会注意的内容。
1.注意python版本
2.注意自己的cann版本等,关注自己是aarch还是x86_64
3.还有一个问题就是要提前安装好别的包,我在这里卡了很久
不确定是哪个包影响的结果,因为我是conda管理的环境,照着文档安装的时候有差错。在前面的安装依赖中还要求了安装这些
pip3 install attrs
pip3 install numpy
pip3 install decorator
pip3 install sympy
pip3 install cffi
pip3 install pyyaml
pip3 install pathlib2
pip3 install psutil
pip3 install protobuf
pip3 install scipy
pip3 install requests
pip3 install absl-py
4.现在在看apex安装的问题,因为这种方式下找不到torch安在哪里了
安装后可以这样看对不对
python -c "import torch;import torch_npu; a = torch.ones(3, 4).npu(); print(a + a);"
应该显示
[[2., 2., 2., 2.],
[2., 2., 2., 2.],
[2., 2., 2., 2.]]