Paddle paddle2.1.0版本GPU显存溢出问题

6yt4nkrj  于 2021-11-30  发布在  Java
关注(0)|答案(2)|浏览(809)

paddle2.1.0版本GPU显存溢出:

用paddle2.1.0版本训练的过程中出现GPU显存溢出的报错,用的环境是aistudio的16G V100环境。报错的截图如下:
图1

这里显存溢出的问题是在我用了下面的tensor切片索引操作后出现的:
图2:

在paddle2.0.2版本中上面的操作是不支持的,所以在2.0.2版本中我做了一个numpy的转换,如下图3,这种方式是可以的,而且不会出现显存溢出问题。
图3:

发布了2.1.0版本后我抱着试一试的态度,试了下图2的写法,发现2.1.0版本支持了这种操作,但是就出现了如图1所示的显存溢出操作,而且看报错信息好像是在反向传播过程中出现的,不清楚到底是什么原因,希望有大佬能帮忙分析一下

mwyxok5s

mwyxok5s1#

您好,我们已经收到了您的问题,会安排技术人员尽快解答您的问题,请耐心等待。请您再次检查是否提供了清晰的问题描述、复现代码、环境&版本、报错信息等。同时,您也可以通过查看官网API文档常见问题历史IssueAI社区来寻求解答。祝您生活愉快~

Hi! We've received your issue and please be patient to get responded. We will arrange technicians to answer your questions as soon as possible. Please make sure that you have posted enough message to demo your request. You may also check out the APIFAQGithub Issue and AI community to get the answer.Have a nice day!

a0x5cqrl

a0x5cqrl2#

请问能否提供一下复现代码呢,这里需要在16G V100上复现一下,定位一下问题在哪里,谢谢!

相关问题