search

diviz-mit / multiduration-saliency

13 stars 3 forks source link

CPU inferencing #2

Closed ghost closed 4 years ago

ghost commented 4 years ago

How to inference using CPU-only mode

a-newman commented 4 years ago

Hi deepseek, thanks for your question. It sounds like you need to configure Keras to use the cpu only. You can do this by installing the tensorflow backend (as opposed to tensorflow-gpu), or you can see this Stack Overflow thread for suggestions of how to force Keras to use the cpu.