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

GPU Memory #18

Closed
tladyman opened this issue Feb 16, 2017 · 4 comments
Closed

GPU Memory #18

tladyman opened this issue Feb 16, 2017 · 4 comments

Comments

@tladyman
Copy link

Hi,

When running rtpose.bin I'm running out of GPU memory.

116 syncedmem.cpp:64] Check failed: error == cudaSuccess (2 vs. 0) out of memory

Is there any way to decrease the amount of memory used or is the answer to use a GPU with more memory?

Cheers,
Tom

@gineshidalgo99
Copy link
Member

gineshidalgo99 commented Feb 16, 2017

Hi, we are working in a version that uses less memory (it should fit in 2GB GPUs). It will be released in around 3-5 weeks. However, if you are using a bigger GPU and it does not fit, you should reduce the number of scales and net resolution to the default values, as well as enable cuDNN. Thanks!

@ZheC
Copy link
Member

ZheC commented Feb 16, 2017

4 Options to reduce memory usage: (1) Using MPI model instead of COCO model (2) Modifying the test.prototxt and reducing the stage number from 6 to 3. (3) Using one scale for testing (4) Using smaller --net_resolution for testing

@ZheC ZheC closed this as completed Mar 8, 2017
@madhawav
Copy link

madhawav commented May 1, 2017

@gineshidalgo99 Is the version for 2GB VGAs available?

@gineshidalgo99
Copy link
Member

I am very sorry for the delay, it will be released within the next 2-3 weeks. Thank you for your patience.

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

4 participants