search

rampasek / GraphGPS

Recipe for a General, Powerful, Scalable Graph Transformer
MIT License
643 stars 114 forks source link

About Multiple GPUs Running GraphGPS in Parallel #13

Closed Ranceeeee closed 1 year ago

Ranceeeee commented 1 year ago

Hello GraphGPS team, have you tried to run GraphGPS with multiple GPUs? If you have tried, what is the specific operation mode? Thank you.

rampasek commented 1 year ago

Hi @Ranceeeee, unfortunately I have not run GraphGPS in a multiple GPU setting. The latest PyG release 2.1.0 has brought PyTorch Lightning support for GraphGym, upgrading to which in the future may be one of the easier ways to enable multiple GPU regime.