-
Notifications
You must be signed in to change notification settings - Fork 620
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
[💡SUG] Will multi-GPU be supported in the next version? #959
Comments
@sxhysj Sorry, considering the complexity of implementation, RecBole's next version still does not support multi-GPU training. We have taken multi-GPU training into consideration. If you have some ideas or relevant papers, comments and suggestions are very welcome as always. |
I suggest you to learn the method of DJL (DeepJavaLibrary https://djl.ai/ ), it could use all the gpus of single machine. |
Any updates on this? |
@sxhysj Sorry, the implementation of this feature is a bit complex in our framework and it still takes some time to add it. BTW, @juyongjiang has given his implementation in #961, which works well for sequential recommendation models. You can refer it and try to modify our code by yourself. |
Hello @sxhysj, we have released the 1.1.1 version of RecBole on Oct 5 2022, and the distributed training and evaluation is supported in our framework now. You can refer to our documentation for more details and running examples. Thank you very much for your attention and kindly advice! The suggestions and feedback from open source users are our driving force for continuous updating. We welcome you very much to use our library again and give your valuable comments. |
Will multi-GPU be supported in the next version?
The text was updated successfully, but these errors were encountered: