华为昇腾pytorch安装踩坑记

本文讲述了在安装Python相关库时遇到的问题,包括Python版本选择、CANN版本确认、使用conda环境下的包安装顺序,以及如何解决Apex安装时找不到torch的问题。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

尝试了几十次最终才差不多,主要总结几个可能不会注意的内容。

1.注意python版本

2.注意自己的cann版本等,关注自己是aarch还是x86_64

查询自己cann包信息

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.]]

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值