About the use of GPU

[DreamerDW]

Hi Ruiqi,
Thanks for your nice work! Does your code support running on GPU? I haven't modified it successfully.

Thank you very much!

[yliuhz]

I solved the problem by simultaneously modifying the "CUDA_VISIBLE_DEVICES" setting in two .py files, namely "clustering.py" and "link_prediction.py", since the main "run.py" file import both of them. I guess that you only modified either one of them.