Return a list of logical devices created by runtime.
See Migration guide for more details.
tf.config.list_logical_devices( device_type=None )
Logical devices may correspond to physical devices or remote devices in the cluster. Operations and tensors may be placed on these devices by using the
name of the
tf.config.list_logical_devices triggers the runtime to configure any
tf.config.PhysicalDevice visible to the runtime, thereby preventing further configuration. To avoid runtime initialization, call
logical_devices = tf.config.list_logical_devices('GPU') if len(logical_devices) > 0: # Allocate on GPU:0 with tf.device(logical_devices.name): one = tf.constant(1) # Allocate on GPU:1 with tf.device(logical_devices.name): two = tf.constant(2)
| ||(optional string) Only include devices matching this device type. For example "CPU" or "GPU".|
| List of initialized |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.