run the classifier on tensorflow GPU

[amapic]

Hi,
Do you know if I can easily change your code to make it work on GPU ?

[ardamavi]

Yes you can with no any change in code, If you using NVIDIA GPU, CUDA and TensorFlow GPU version.
Verify your GPU with K.tensorflow_backend._get_available_gpus() code (But first, you must import keras.backend).

If you want to use with multi GPU, import keras.utils.multi_gpu_model module and add model = multi_gpu_model(model) line to code.

[amapic]

Thank you !