Out of Memory error with GPU having 80% of its vram free.

autonomousvision / stylegan-xl

[SIGGRAPH'22] StyleGAN-XL: Scaling StyleGAN to Large Diverse Datasets

MIT License

961 stars 113 forks source link

Out of Memory error with GPU having 80% of its vram free. #90

Closed nom57 closed 2 years ago

nom57 commented 2 years ago

I'm getting OOM errors even at lowest resolution batch_size = 1 , with 12gb vram and 32gb system ram. Strangely the OOM error message says I had more than 9.7gb of vram left , yet I got OOM error'ed for much smaller allocations.

nom57 commented 2 years ago

found the solution , needed to use CUDA 11.7