-
Notifications
You must be signed in to change notification settings - Fork 88
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 Transfer #65
Comments
@mughetto @benedekrozemberczki A simple option for flexible GPU support is to use Accelerate. I played around with a quick implementation for chemicalx and it's very easy, only changed a few lines of code. I tested on CPU and single GPU and it works well so far. GPU speeds up training and gets high utilisation, e.g the deepddi example is so much faster with GPU enabled 🏎. If you're interested in this approach I can make an initial PR? |
@GavEdwards Sounds great! Right know, we really struggle with speed for the experiments. Can you make sure that the molecule and fingerprint based models both work on GPU? |
draft PR: #76 |
The text was updated successfully, but these errors were encountered: