TextSum icon indicating copy to clipboard operation
TextSum copied to clipboard

Published 20 hours ago • verified publisher icon xiaoyu2018

请问batch_size=8，仍有这个问题

Open ghost opened this issue 3 years ago • 3 comments

CUDA out of memory. Tried to allocate 512.00 MiB (GPU 0; 11.91 GiB total capacity; 10.84 GiB already allocated; 205.62 MiB free; 10.87 GiB reserved in total by PyTorch)

ghost avatar Apr 20 '22 09:04 ghost

但是显示的0号GPU并没有进程 23% 36C P8 11W / 250W | 0MiB / 12194MiB | 0% Default | | | | N/A |

ghost avatar Apr 20 '22 09:04 ghost

你把batch size 调成1试试

xiaoyu2018 avatar Apr 20 '22 10:04 xiaoyu2018

呜呜呜batchsize设成1是可以的！看来我的GPU还是太垃圾了

ghost avatar Apr 20 '22 10:04 ghost