Python neural network with Keras runs on CPU, but crashes on the GPU

Question

I implemented a neural network that learns to play PacMan using gym,box2d and gym[atari] with Keras models. The training was very slow so I tried to make in run on my GTX 10