How do I check if keras is using gpu version of tensorflow?

by

You are using the GPU version. You can list the available tensorflow devices with (also check this question):

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices()) # list of DeviceAttributes

EDIT:

With tensorflow >= 1.4 you can run the following function:

import tensorflow as tf
tf.test.is_gpu_available() # True/False

# Or only check for gpu's with cuda support
tf.test.is_gpu_available(cuda_only=True)

EDIT 2:

The above function is deprecated in tensorflow > 2.1. Instead you should use the following function:

import tensorflow as tf
tf.config.list_physical_devices('GPU')

NOTE:

In your case both the cpu and gpu are available, if you use the cpu version of tensorflow the gpu will not be listed. In your case, without setting your tensorflow device (with tf.device("..")), tensorflow will automatically pick your gpu!

In addition, your sudo pip3 list clearly shows you are using tensorflow-gpu. If you would have the tensoflow cpu version the name would be something like tensorflow(1.1.0).

Check this issue for information about the warnings.

More Related Contents:

  1. Keras + Tensorflow and Multiprocessing in Python
  2. Negative dimension size caused by subtracting 3 from 1 for ‘conv2d_2/convolution’
  3. Loading a trained Keras model and continue training
  4. Multiple outputs in Keras
  5. Keras Dense layer’s input is not flattened
  6. TensorFlow Only running on 1/32 of the Training data provided [duplicate]
  7. How to export Keras .h5 to tensorflow .pb?
  8. How to choose cross-entropy loss in TensorFlow?
  9. Tensorflow – ValueError: Failed to convert a NumPy array to a Tensor (Unsupported object type float)
  10. Keras Sequential model input layer
  11. How to concatenate two layers in keras?
  12. Proper way to feed time-series data to stateful LSTM?
  13. How to use a Keras RNN model to forecast for future dates or events?
  14. TypeError: __array__() takes 1 positional argument but 2 were given (Image classification Keras)
  15. Tensorflow 2.0 – AttributeError: module ‘tensorflow’ has no attribute ‘Session’
  16. How does Keras 1d convolution layer work with word embeddings – text classification problem? (Filters, kernel size, and all hyperparameter)
  17. ValueError at /image/ Tensor Tensor(“activation_5/Softmax:0”, shape=(?, 4), dtype=float32) is not an element of this graph
  18. What do I need K.clear_session() and del model for (Keras with Tensorflow-gpu)?
  19. Passing output of a CNN to BILSTM
  20. Why can’t I get reproducible results in Keras even though I set the random seeds?
  21. What is the role of TimeDistributed layer in Keras?
  22. Tensorflow : logits and labels must have the same first dimension
  23. Tensor is not an element of this graph
  24. How to create a keras layer with a custom gradient in TF2.0?
  25. Custom ImageDataGenerator() for half a million images where labels and pixels are in 2 separate DataFrames using Keras (or any other library) [closed]
  26. NotImplementedError: Cannot convert a symbolic Tensor (2nd_target:0) to a numpy array
  27. Tensorflow One Hot Encoder?
  28. Can I send callbacks to a KerasClassifier?
  29. How to use TensorFlow metrics in Keras
  30. How to visualize RNN/LSTM gradients in Keras/TensorFlow?

Leave a Comment