RuntimeError: CUDA out of memory. Tried to allocate 1024.00 MiB (GPU 1; 10.76 GiB total capacity; 4.05 GiB already allocated; 910.69 MiB free; 5.38 GiB reserved in total by PyTorch)

wyhsirius / g3an-project

[CVPR 20] G3AN: Disentangling Appearance and Motion for Video Generation

https://wyhsirius.github.io/G3AN/

MIT License

38 stars 8 forks source link

RuntimeError: CUDA out of memory. Tried to allocate 1024.00 MiB (GPU 1; 10.76 GiB total capacity; 4.05 GiB already allocated; 910.69 MiB free; 5.38 GiB reserved in total by PyTorch) #2

Closed DreamMr closed 3 years ago

DreamMr commented 3 years ago

Hello ! I using "--use_attention", CUDA out of memory occurs when I train for around 400 steps. And not use "--use_attention", I found that memory get bigger and bigger.

DreamMr commented 3 years ago

That's my fault. Sorry !