how to use gpu
hxtsch opened this issue · 2 comments
hxtsch commented
thanks. I'm inspired by the code.
bapalto commented
Using a GPU is a question of setting the backend (Theano or TensorFlow) of Keras properly. If you are using TensorFlow as a backend please pay attention to "pip install tensorflow-gpu".
hxtsch commented
Thanks for your reply.
I have solved the problem.
Thank you.