Can I run Keras model on gpu?

by

Yes you can run keras models on GPU. Few things you will have to check first.

  1. your system has GPU (Nvidia. As AMD doesn’t work yet)
  2. You have installed the GPU version of tensorflow
  3. You have installed CUDA installation instructions
  4. Verify that tensorflow is running with GPU check if GPU is working

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

for TF > v2.0

sess = tf.compat.v1.Session(config=tf.compat.v1.ConfigProto(log_device_placement=True))

(Thanks @nbro and @Ferro for pointing this out in the comments)

OR

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

output will be something like this:

[
  name: "/cpu:0"device_type: "CPU",
  name: "/gpu:0"device_type: "GPU"
]

Once all this is done your model will run on GPU:

To Check if keras(>=2.1.1) is using GPU:

from keras import backend as K
K.tensorflow_backend._get_available_gpus()

All the best.

More Related Contents:

  1. Keras Dense layer’s input is not flattened
  2. How to export Keras .h5 to tensorflow .pb?
  3. Tensorflow – ValueError: Failed to convert a NumPy array to a Tensor (Unsupported object type float)
  4. Keras + Tensorflow and Multiprocessing in Python
  5. Proper way to feed time-series data to stateful LSTM?
  6. How to use a Keras RNN model to forecast for future dates or events?
  7. TypeError: __array__() takes 1 positional argument but 2 were given (Image classification Keras)
  8. Tensorflow 2.0 – AttributeError: module ‘tensorflow’ has no attribute ‘Session’
  9. How does Keras 1d convolution layer work with word embeddings – text classification problem? (Filters, kernel size, and all hyperparameter)
  10. ValueError at /image/ Tensor Tensor(“activation_5/Softmax:0”, shape=(?, 4), dtype=float32) is not an element of this graph
  11. What do I need K.clear_session() and del model for (Keras with Tensorflow-gpu)?
  12. expected ndim=3, found ndim=2
  13. Passing output of a CNN to BILSTM
  14. How to get other metrics in Tensorflow 2.0 (not only accuracy)?
  15. Removing then Inserting a New Middle Layer in a Keras Model
  16. Why can’t I get reproducible results in Keras even though I set the random seeds?
  17. AttributeError: module ‘tensorflow.python.keras.utils.generic_utils’ has no attribute ‘populate_dict_with_module_objects’
  18. How do I check if keras is using gpu version of tensorflow?
  19. Tensorflow : logits and labels must have the same first dimension
  20. Keras LSTM input dimension setting
  21. How to create a keras layer with a custom gradient in TF2.0?
  22. How to fix “AttributeError: module ‘tensorflow’ has no attribute ‘get_default_graph'”?
  23. ‘Tensor’ object has no attribute ‘lower’
  24. Unexpected keyword argument ‘ragged’ in Keras
  25. ValueError: Output tensors to a Model must be the output of a TensorFlow `Layer`
  26. Strange behaviour of the loss function in keras model, with pretrained convolutional base
  27. TensorFlow 2.0 Keras: How to write image summaries for TensorBoard
  28. Multi-input Multi-output Model with Keras Functional API
  29. Invalid Argument Error / Graph Execution Error
  30. How to fix ipykernel_launcher.py: error: unrecognized arguments in jupyter?

Leave a Comment