I have Keras installed with the Tensorflow backend and CUDA. I'd like to sometimes on demand force Keras to use CPU. Can this be done without say installing a separate CPU-only Tensorflow in a virtual environment? If so how? If the backend were Theano, the flags could be set, but I have not heard of Tensorflow flags accessible via Keras.
Python – Can Keras with Tensorflow backend be forced to use CPU or GPU at will
kerasmachine learningpythontensorflow
Best Answer
If you want to force Keras to use CPU
Way 1
before Keras / Tensorflow is imported.
Way 2
Run your script as
See also