search

PeterL1n / RobustVideoMatting

Robust Video Matting in PyTorch, TensorFlow, TensorFlow.js, ONNX, CoreML!
https://peterl1n.github.io/RobustVideoMatting/
GNU General Public License v3.0
8.61k stars 1.14k forks source link

CUDA out of memory #64

Closed antithing closed 3 years ago

antithing commented 3 years ago

I am trying to train a model, and in stage 2 I get the following error:

RuntimeError: CUDA out of memory. Tried to allocate 1.71 GiB (GPU 0; 8.00 GiB total capacity; 3.55 GiB already allocated; 289.88 MiB free; 6.22 GiB reserved in total by PyTorch)

I have set the num_workers to 1 I am using a GTX 1080 card.

Am I unable to train on this card? Or can i adjust some parameters somewhere?

Thanks!

PeterL1n commented 3 years ago

We trained on 4 V100 each with 32G VRAM. It's not gonna work on GTX 1080.

antithing commented 3 years ago

Understood. Thanks!