deepchem
deepchem copied to clipboard
Enhance GPU Support in DeepChem
This PR introduces a feature to automatically detect GPU availability and use it for computations within DeepChem models, enhancing performance and efficiency. By ensuring that GPU resources are utilized when available, this feature can significantly speed up model training and inference processes on suitable hardware.
Type of Change:
- [ ] Bug fix (non-breaking change which fixes an issue)
- [x] New feature (non-breaking change which adds functionality)
- [ ] Breaking change (fix or feature that would cause existing functionality to not work as expected)
- [ ] Documentation update
Checklist
- [x] My code follows the code style of this project.
- [x] I have added tests to cover my changes.
- [x] All new and existing tests passed.
Hey @mirsadra Can you join in Office Hour [1], there we can discuss about this PR.
Ref
[1].. https://forum.deepchem.io/t/announcing-the-deepchem-office-hours/293