Train TensorFlow models faster with GPU