Google collab keeps crashing

[ecoy]

Hi

I’m training a model on google colab and it keeps crashing - from the logs this is what I see

Jul 20, 2023, 7:25:12 PM	WARNING	[bt] (2) /usr/local/lib/python3.10/dist-packages/dgl/libdgl.so(dgl::runtime::CUDADeviceAPI::FreeWorkspace(DGLContext, void*)+0x56) [0x7ca2d5a69f06]
Jul 20, 2023, 7:25:12 PM	WARNING	[bt] (1) /usr/local/lib/python3.10/dist-packages/dgl/libdgl.so(dgl::runtime::WorkspacePool::FreeWorkspace(DGLContext, void*)+0x1cb) [0x7ca2d58eb6fb]

I’m using the T4 GPU and already pro+. I dont get any other errors with the code.

This is how I import dgl

!pip install dgl==1.0.1+cu117 -f https://data.dgl.ai/wheels/cu117/repo.html

Please let me know if you have any ideas

[peizhou001]

Hi @ecoy, your provided is message warning, which should not be the root cause of the crash. Could you post the error message and give a reproducible script?

[ecoy]

it happens when I try to run PinSAGE on google collab