Я столкнулся с этой ошибкой: RuntimeError: CUDA недостаточно памяти. Попытка выделить 116,00 МБ (GPU 0; общая емкость 7,8 ГБ; 1,95 ГБ уже выделено; 110,00 МБ бесплатно; всего 2,00 ГБ зарезервировано PyTorch)
Однако размер пакета уже очень мал, поэтому я не знаю в чем проблема.
Я работаю в сети Mask RCNN.
Когда я использую nvidia-smi, чтобы посмотреть на мой графический процессор, это вывод. : | NVIDIA-SMI 440.59 Версия драйвера: 440.59 CUDA Версия: 10.2 | | ------------------------------- + ----------------- ----- + ---------------------- + | Название графического процессора Персистент-М | Bus-Id Disp.A | Летучий Uncorr. E CC | | Fan Temp Perf Pwr: Использование / Крышка | Использование памяти | GPU-Util Compute M. |
| 0 GeForce RTX 2070 Off | 00000000: 01: 00.0 выкл. | N / A | | 20% 35 C P8 8 Вт / 175 Вт | 1260MB / 7982MiB | 0% по умолчанию | + ------------------------------- + ----------------- ----- + ---------------------- +
+ -------------- -------------------------------------------------- ------------- + | Процессы: Память GPU | | GPU PID Тип Имя процесса Использование | | 0 5282 C ... / anaconda3 / envs / deeplearning / bin / python 1249MiB | + ------------------------------------------------- ---------------------------- +
Спасибо.