Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

RuntimeError: CUDA out of memory #46

Closed
ghost opened this issue Mar 28, 2021 · 1 comment
Closed

RuntimeError: CUDA out of memory #46

ghost opened this issue Mar 28, 2021 · 1 comment

Comments

@ghost
Copy link

ghost commented Mar 28, 2021

I've found a fix, it's below.

First of all, thank you for this wonderful tool!

I have gtx 750ti GPU, and I'm getting this error (running python interactive.py --model=StyleGAN2 --class=ffhq --layer=style --use_w -n=1_000_000 -b=10_000). I'm using Linux Mint 20. I've also tried to setup StyleGAN2. When I run the commands from "Usage", it opens the window and then closes it when I move any slider. Apparently, 2 GB of VRAM is not enough. Is there any way to make it work with ganspace? For example reduce batch size? Or use CPU instead? If not, are there google colab notebooks that can be used with the same functionality?

P.S. I was able to run it after changing -b option to 1_000. I assume it's the batch size.

@ghost ghost closed this as completed Mar 28, 2021
@FlashlightET
Copy link

same here, but 1660S and windows 10 even with -b on 1,000. and i have 6gb of vram!!! why does the 2gb one work???

This issue was closed.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant