Bert-Chinese-Text-Classification-Pytorch icon indicating copy to clipboard operation
Bert-Chinese-Text-Classification-Pytorch copied to clipboard

Published 20 hours ago • verified publisher icon 649453932

RuntimeError: CUDA out of memory. Tried to allocate 48.00 MiB (GPU 0; 6.00 GiB total capacity; 4.58 GiB already allocated; 37.63 MiB free; 51.94 MiB cached)2

Open Prometheus-F opened this issue 3 years ago • 2 comments

请问出现这个怎么解决呢

Prometheus-F avatar Dec 04 '21 08:12 Prometheus-F

減少batch_size 結案

ShunKetsuRyou avatar Dec 08 '21 01:12 ShunKetsuRyou

没显存了，把batch_size降低； 举个例子：batch_size = 128 改成batch_size = 64

HANjf0822 avatar Aug 29 '22 07:08 HANjf0822