在Atlas200下部署torch学习框架
收藏回复举报
在Atlas200下部署torch学习框架
t('forum.solved') 已解决
发表于2024-06-23 19:09:21
0 查看

【用户类型】:企业

【昇腾产品型号】:Atlas 200 DK

【版本信息】:

--操作系统版本:ubuntu.aarch64

--固件驱动版本:npu-smi 21.0.4 Version: 21.0.4

--CANN版本:CANN 6.0.1

--Pytorch版本:1.8.1

--Python版本:Python3.7.5

【官方样例or个人代码】:

--参考官方文档:hhttps://www.hiascend.com/document/detail/zh/canncommercial/601/envdeployment/instg/instg_000035.html

【开发内容】:部署开发环境,试图迁移自研的torch项目,用作推理

【问题描述】:目前部署过程中,先是安装了torch-1.11.0,但是无法提示不适配310处理器,后面安装了torch-1.8.1,但是出现下面问题。请问怎么解决呢?

Traceback (most recent call last):
  File "/home/HwHiAiUser/miniconda3/envs/asrpy37/lib/python3.7/site-packages/torch_npu/__init__.py", line 29, in <module>
    import torch_npu.npu
  File "/home/HwHiAiUser/miniconda3/envs/asrpy37/lib/python3.7/site-packages/torch_npu/npu/__init__.py", line 42, in <module>
    from .utils import (is_initialized, _lazy_call, _lazy_init, init, set_dump,
  File "/home/HwHiAiUser/miniconda3/envs/asrpy37/lib/python3.7/site-packages/torch_npu/npu/utils.py", line 27, in <module>
    import torch_npu._C
ImportError: /home/HwHiAiUser/miniconda3/envs/asrpy37/lib/python3.7/site-packages/torch_npu/lib/libtorch_npu.so: undefined symbol: _ZNK5torch8autograd4Node4nameB5cxx11Ev
Segmentation fault (core dumped)

本帖最后由 匿名用户2024/06/27 16:35:38 编辑

我要发帖子