I have the same problem. tensorflow.config.set_soft_device_placement(True) should solve such a problem, but it did not.