Can Keras with Tensorflow backend be forced to use CPU or GPU at will?

by

If you want to force Keras to use CPU

Way 1

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"   # see issue #152
os.environ["CUDA_VISIBLE_DEVICES"] = ""

before Keras / Tensorflow is imported.

Way 2

Run your script as

$ CUDA_VISIBLE_DEVICES="" ./your_keras_code.py

See also

  1. https://github.com/keras-team/keras/issues/152
  2. https://github.com/fchollet/keras/issues/4613

More Related Contents:

  1. Keras Dense layer’s input is not flattened
  2. Make a custom loss function in keras
  3. Keras not training on entire dataset
  4. Deep-Learning Nan loss reasons
  5. How to apply gradient clipping in TensorFlow?
  6. Proper way to feed time-series data to stateful LSTM?
  7. ValueError at /image/ Tensor Tensor(“activation_5/Softmax:0”, shape=(?, 4), dtype=float32) is not an element of this graph
  8. Keras custom loss function: Accessing current input pattern
  9. How to get other metrics in Tensorflow 2.0 (not only accuracy)?
  10. How can I use a pre-trained neural network with grayscale images?
  11. Removing then Inserting a New Middle Layer in a Keras Model
  12. How to know if underfitting or overfitting is occuring?
  13. Save and load model optimizer state
  14. Unknown initializer: GlorotUniform when loading Keras model
  15. How to create a keras layer with a custom gradient in TF2.0?
  16. ‘Tensor’ object has no attribute ‘lower’
  17. ValueError: Output tensors to a Model must be the output of a TensorFlow `Layer`
  18. How to fix “ResourceExhaustedError: OOM when allocating tensor”
  19. How is the Keras Conv1D input specified? I seem to be lacking a dimension
  20. What are logits? What is the difference between softmax and softmax_cross_entropy_with_logits?
  21. Keras + Tensorflow and Multiprocessing in Python
  22. How to concatenate two layers in keras?
  23. Does Any one got “AttributeError: ‘str’ object has no attribute ‘decode’ ” , while Loading a Keras Saved Model
  24. ValueError: Error when checking target: expected model_2 to have shape (None, 252, 252, 1) but got array with shape (300, 128, 128, 3)
  25. TensorFlow operator overloading
  26. expected ndim=3, found ndim=2
  27. Why can’t I get reproducible results in Keras even though I set the random seeds?
  28. AttributeError: module ‘tensorflow.python.keras.utils.generic_utils’ has no attribute ‘populate_dict_with_module_objects’
  29. What is the role of TimeDistributed layer in Keras?
  30. Tensor is not an element of this graph

Leave a Comment