atlas 200i dk a2的310b4芯片是否支持torch-npu、BrokenPipeError
收藏回复举报
atlas 200i dk a2的310b4芯片是否支持torch-npu、BrokenPipeError
t('forum.solved') 已解决
发表于2024-06-09 12:16:57
0 查看

 在镜像烧录之后,安装了对应版本的torch和torch-npu

分别是2.1.0版本的2.1.0.rc1版本的torch-npu,出现了如下报错,因此更换了torch-npu的版本为2.1.0.post3,解决了如下问题,但出现了新的疑问。

在将tensor放到npu上进行运算时,出现了BrokenPipeError。

下图是测试代码:

cke_26618.png

下图是报错信息:

我要发帖子