我怎么知道Pytorch中的函数是否分配了新内存?

我怎么知道Pytorch中的函数是否分配了新内存?,pytorch,Pytorch,最近,我陷入了一种情况,在我的模型中,输入数据确实消耗了大量内存。当我在网络层中操作数据时,这会导致大量内存使用。我很想知道这些操作是否会分配新的内存块。我看到pytorch文档只找到了如何使用该函数。我想知道是否有一个医生或一些网站或任何其他官方帮助我。例如,view、permute或Continental等函数是否会分配新内存块,您如何知道这一点。它真的吸引了我,谢谢你的帮助。如果你在GPU上运行,检查内存消耗的最好方法是使用linux命令nvidia smi。你可以在jupyter笔记本中

最近,我陷入了一种情况,在我的模型中,输入数据确实消耗了大量内存。当我在网络层中操作数据时,这会导致大量内存使用。我很想知道这些操作是否会分配新的内存块。我看到pytorch文档只找到了如何使用该函数。我想知道是否有一个医生或一些网站或任何其他官方帮助我。例如,view、permute或Continental等函数是否会分配新内存块,您如何知道这一点。它真的吸引了我,谢谢你的帮助。

如果你在GPU上运行,检查内存消耗的最好方法是使用linux命令nvidia smi。你可以在jupyter笔记本中使用!英伟达smi。这样,在执行任何Pytorch命令后,您都可以检查是否分配了新内存

谢谢您的帮助。但我想,在一份文件或其他任何文件中,我想知道的是清楚的说明。我认为这可能对我们的网络设计有很大帮助。