GPU support within stellargraph


I’m interested in working with GraphSAGE, and seeing that StellarGraph’s implementations are based on keras, I was wondering if using GPUs is as simple as installing a GPU supported version of keras and tensorflow.



thank you for your interest to StellarGraph and welcome to our community.

You should be able to use GPU acceleration without much effort assuming that you have the correct drivers and libraries installed for your hardware.

On our GPU workstation, we can train with GPU by doing the following.

First, uninstall tensorflow (pip uninstall tensorflow) and then install tensorflow-gpu (pip install tensorflow-gpu).

That’s it. Training your model should now utilise your GPU.

Let us know if this doesn’t work.