About the use of GPU

Open DreamerDW opened this issue 5 years ago • 1 comments

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

Thank you very much!

May 22 '20 23:05 DreamerDW

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.

Nov 07 '22 07:11 yliuhz