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

Running test net on a separate device id #6657

Open
ishan193 opened this issue Jan 5, 2019 · 0 comments
Open

Running test net on a separate device id #6657

ishan193 opened this issue Jan 5, 2019 · 0 comments

Comments

@ishan193
Copy link

ishan193 commented Jan 5, 2019

Hi,
I am trying to find a way to separate out my test net and use a specific gpu id for all test related stuff. The need occurred because sometimes the testing at a specific interval during training is impossible if the net is very big. If we don't allocate the diff and data on one particular device (gpu), then we can just use if for forward prop during testing.
Currently the solver initializes the test net and if I hack there and set a particular device id, then the NCCL is not able to use the remaining gpus for multi-gpu training properly.

Any advice or quick hack?

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