search

PeterL1n / BackgroundMattingV2

Real-Time High-Resolution Background Matting
MIT License
6.85k stars 953 forks source link

CUDA out of memory #142

Closed edbea closed 3 years ago

edbea commented 3 years ago

你好， RuntimeError: CUDA out of memory. Tried to allocate 72.00 MiB (GPU 0; 1024.00 MiB total capacity; 378.05 MiB already allocated; 0 bytes free; 418.00 MiB reserved in total by PyTorch)

除了更换更高更强的GPU，有无其他的解决方案，谢谢

PeterL1n commented 3 years ago

1G显存确实有点小，你是做inference还是training？

PeterL1n commented 3 years ago

Closing due to inactivity.