Problem transferring graphs to GPU on Colab

Hi! I’m using DGL on Colab.

My installation is as follows:

image

I tried to create a graph and transfer it to a GPU. However, after calling .to(device), the graph still stays on CPU.

image

I have tried both DGL 0.6 and DGL 0.7. Does anyone know a solution to this problem? I’m new to DGL so I have a feeling the problem is due to some dumb mistake on my part but so far I cannot figure out what’s wrong.

Try

g = g.to(device)

This topic was automatically closed 30 days after the last reply. New replies are no longer allowed.