tf.config.set_visible_devices