tf.config.set_visible_devices

Set the list of visible devices.