Out of memory error on GPU 0. Cannot allocate 90.250244MB memory on GPU 0, available memory is only 11.450000MB.
您好，具体的项目URL方便提供下吗？
您好。我们遇到了跟您类似的问题，请问您最后解决了吗
你好，遇到同样的问题了，请问解决了没有
可能用的时候不够
loss.backward()算梯度时候显存占用有点大。我碰到过第一个batch可以更新参数，到第二个batch算梯度时候报显存不足。
调小batch_size不就行了
把batch size调小即可，因为他有个预分配的过程会判断显存够不够
这个就是显存不够
解决方法楼下也说了，batchsize要调小
反正就是数据太大，大到GPU塞不下
除了batch还可以缩减网络、减小训练数据规模...
楼主问题解决了嘛
您好，具体的项目URL方便提供下吗？
您好。我们遇到了跟您类似的问题，请问您最后解决了吗
你好，遇到同样的问题了，请问解决了没有
可能用的时候不够
loss.backward()算梯度时候显存占用有点大。我碰到过第一个batch可以更新参数，到第二个batch算梯度时候报显存不足。
调小batch_size不就行了
把batch size调小即可，因为他有个预分配的过程会判断显存够不够
这个就是显存不够
解决方法楼下也说了，batchsize要调小
反正就是数据太大，大到GPU塞不下
除了batch还可以缩减网络、减小训练数据规模...
楼主问题解决了嘛