cuda out of memory
cuixintong opened this issue · comments
我用作者提供的代码在我的设备上运行main.py,报错内存不足，只有当我把batchsize设置为1时才能运行成功，但是我用的时TeslaV100 32G显卡，作者用的是2080Ti，作者设置的batchsize是16，为什么我的会内存不足呢，有同学遇到这个问题吗
你好，我也遇到了这个问题，请问你已经解决了吗
没有解决，不知道为啥
好的，还是非常感谢
w = w.view(-1, self.gps, self.dim)[:, :, :, None, None]，我感觉应该是这里的问题，这里把w的维度变成了5维，然后batch_size就非常大了，但是我也不知道要怎么去改，不过也不一定是这个原因，只是我的猜想
不清楚呀，这个会影响吗，我好奇的是我觉得我的这个32G卡已经挺大了，为什么作者能跑起来 我这不行呢
这个问题可能是图像大小的问题，看看图像大小是否为240，我是因为这个原因后来解决了