-
Notifications
You must be signed in to change notification settings - Fork 18.7k
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
multi gpu support for lstm layer #4851
Comments
Does it work with the nccl version? On Oct 17, 2016 8:41 AM, "zachluo" notifications@github.com wrote:
|
this PR addresses the problem. |
as @junshi15 says, it works ! |
Note this is solved by further releases in the switch to NCCL #4563, which is included in 1.0. |
The implement of lstm layer in caffe is to use a sub-net. But it seems to fail to use multi gpu when training lstm. It outputs:
Check failed: Caffe::root_solver() || root_net_ root_net_ needs to be set for all non-root solvers
Any one meet this issue?
The text was updated successfully, but these errors were encountered: