cuda out of memory

[cuixintong]

我用作者提供的代码在我的设备上运行main.py,报错内存不足，只有当我把batchsize设置为1时才能运行成功，但是我用的时TeslaV100 32G显卡，作者用的是2080Ti，作者设置的batchsize是16，为什么我的会内存不足呢，有同学遇到这个问题吗

[Marinerjl]

你好，我也遇到了这个问题，请问你已经解决了吗

[cuixintong]

没有解决，不知道为啥

[Marinerjl]

好的，还是非常感谢

[Marinerjl]

w = w.view(-1, self.gps, self.dim)[:, :, :, None, None]，我感觉应该是这里的问题，这里把w的维度变成了5维，然后batch_size就非常大了，但是我也不知道要怎么去改，不过也不一定是这个原因，只是我的猜想

[cuixintong]

不清楚呀，这个会影响吗，我好奇的是我觉得我的这个32G卡已经挺大了，为什么作者能跑起来 我这不行呢

[Marinerjl]

这个问题可能是图像大小的问题，看看图像大小是否为240，我是因为这个原因后来解决了

[ZhuLiangyu123]

我用作者提供的代码在我的设备上运行main.py,报错内存不足，只有当我把batchsize设置为1时才能运行成功，但是我用的时TeslaV100 32G显卡，作者用的是2080Ti，作者设置的batchsize是16，为什么我的会内存不足呢，有同学遇到这个问题吗
你好，我也遇到了这个问题，请问你解决了吗？