nju-websoft / MultiKE

Multi-view Knowledge Graph Embedding for Entity Alignment, IJCAI 2019
MIT License
114 stars 29 forks source link

Tensorflow CPU #12

Open nikofan18 opened 4 years ago

nikofan18 commented 4 years ago

Hello,

congratulations for your work. Is there any option to use your code at a machine without gpu ? When I am trying to do this, tensorflow uses only one core.

Thank you in advance

sunzequn commented 4 years ago

Hi,

Thanks for your interest in our work. The code can run on a machine with only CPUs. You can use "CUDA_VISIBLE_DEVICES=-1 python xxx.py" to run the code. If you encounter errors, please tell me the details such that I can further fix the code.

nikofan18 commented 4 years ago

Hello, I used CUDA_VISIBLE_DEVICES=-1 to disable the GPU. However, the problem is that your code runs on a single core and my machine has 38 cores.

sunzequn commented 4 years ago

Sorry that the current code is single-process and cannot use multiple CPU cores.

nikofan18 commented 4 years ago

Is there a way to fix it ?