MindSpore张量运算失败报错RuntimeError:Malloc for kernel output failed, Memory isn’t enough, node:Default/Add-op0
收藏回复举报
MindSpore张量运算失败报错RuntimeError:Malloc for kernel output failed, Memory isn’t enough, node:Default/Add-op0
发表于2023-04-20 10:25:08
0 查看

1 系统环境

硬件环境(Ascend/GPU/CPU): GPU

MindSpore版本: mindspore=1.10.1

执行模式(PyNative/ Graph):PyNative

Python版本: Python=3.7.5

操作系统平台: linux

2 报错信息

2.1 问题描述

在python的交互界面下运行脚本,环境显存为24G,在执行print(ms.ops.add(a,b))语句前显存占用为0,在执行完毕print(ms.ops.add(a,b))语句后,抛出错误,显存占用为18647M,未得到释放。

2.2 报错信息

RuntimeError:Malloc for kernel output failed, Memory isn’t enough, node:Default/Add-op0

2.3 脚本代码

import mindspore as ms
import numpy as np

ms.set_context(device_target="GPU")
a=ms.Tensor(np.ones([999,999,999]))
b=ms.Tensor(np.ones([999,999,999]))

print(ms.ops.add(a,b))

3 根因分析

******此处由用户补充详细的定位过程******

考虑代码的执行方式

4 解决方案

******此处由用户填写******

包含文字方案和最终脚本代码

支持上传附件

我要发帖子