seongjunyun/Graph_Transformer_Networks

How much GPU memory is required?

kepsail opened this issue · 2 comments

kepsail commented

My GPU has 10.92 GiB total memory. I run main_sparse.py on ACM data. It raised RuntimeError: CUDA out of memory

iceshzc commented

I get the same issue, and I run the source code on CPU with 32GB, there's no other errors, just a slow speed.

seongjunyun commented

Hi all,
We also ran our code on CPU when datasets were DBLP, ACM.
So we recommended you to run main.py on DBLP and ACM data.
Thank you!

  • 👎5