Yes you can run keras models on GPU. Few things you will have to check first.
- your system has GPU (Nvidia. As AMD doesn’t work yet)
- You have installed the GPU version of tensorflow
- You have installed CUDA installation instructions
- 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.