MultiKE icon indicating copy to clipboard operation
MultiKE copied to clipboard

Tensorflow CPU

Open nikofan18 opened this issue 3 years ago • 4 comments

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

nikofan18 avatar Oct 28 '20 23:10 nikofan18

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.

sunzequn avatar Nov 02 '20 03:11 sunzequn

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.

nikofan18 avatar Nov 02 '20 11:11 nikofan18

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

sunzequn avatar Nov 02 '20 11:11 sunzequn

Is there a way to fix it ?

nikofan18 avatar Nov 02 '20 11:11 nikofan18