CUDA из памяти - PullRequest
       21

CUDA из памяти

0 голосов
/ 30 марта 2020

Я получаю сообщение об ошибке при попытке запустить модель BERT для задачи NER. «CUDA не хватает памяти. Попытка выделить 20,00 МБ (GPU 0; 3,82 ГБ общей емкости; 2,58 ГБ уже выделено; 25,38 МБ свободной; 6,33 МБ кэшировано). Я также пытался уменьшить размер пакета до 1 c

* 1002. *

Прилагается вывод nvidia-smi: -

nvidia-smi

...