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

out of memory #27

Closed
GitUser200607 opened this issue Aug 5, 2017 · 1 comment
Closed

out of memory #27

GitUser200607 opened this issue Aug 5, 2017 · 1 comment

Comments

@GitUser200607
Copy link

I try to run:
python ideepcolor.py --gpu 0
But I get the error:
Setting ab cluster centers in layer: pred_ab
Setting upsampling layer kernel: pred_313_us
test_imgs/mortar_pestle.jpg
scale = 2.000000
F0805 19:15:20.179786 3421 syncedmem.cpp:71] Check failed: error == cudaSuccess (2 vs. 0) out of memory
*** Check failure stack trace: ***
Aborted

I have GTX 660 with 2GB of vram. Is it not enough? What graphic card do I need to run ideepcolor?

@HenkPoley
Copy link

HenkPoley commented Nov 7, 2017

Apparently 3GB is not enough either: #9

There they mention that running on CPU it takes 4.7GB. Of course a part of that is python all the the support libraries. But it should give sort of ballpark requirements.

Here it's mentioned that it used 4.5GB of GPU memory, for a specific example file: #19

There is also a potential fix here: #10 (comment)

@junyanz junyanz closed this as completed Nov 22, 2017
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

3 participants