TensorFlow es una biblioteca Python de código abierto diseñada por Google para desarrollar modelos de aprendizaje automático y redes neuronales de aprendizaje profundo.
device() se usa para especificar explícitamente el dispositivo en el que se debe realizar la operación.
Sintaxis: tensorflow.device( nombre_de_dispositivo )
Parámetros:
- device_name: Especifica el nombre del dispositivo que se utilizará en este contexto.
Devoluciones: devuelve un administrador de contexto que especifica el dispositivo predeterminado que se usará para las operaciones recién creadas.
Ejemplo 1:
Python3
# Importing the library import tensorflow as tf # Initializing Device Specification device_spec = tf.DeviceSpec(job ="localhost", replica = 0, device_type = "CPU") # Printing the DeviceSpec print('Device Spec: ', device_spec.to_string()) # Enabling device logging tf.debugging.set_log_device_placement(True) # Specifying the device with tf.device(device_spec): a = tf.constant([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]]) b = tf.constant([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]]) c = tf.matmul(a, b)
Producción:
Device Spec: /job:localhost/replica:0/device:CPU:* Executing op MatMul in device /job:localhost/replica:0/task:0/device:CPU:0
Ejemplo 2: en esta especificación de dispositivo de ejemplo, se especifica la GPU que se utilizará, pero el sistema no pudo encontrar la GPU, por lo que ejecutará la operación en la CPU.
Python3
# Importing the library import tensorflow as tf # Initializing Device Specification device_spec = tf.DeviceSpec(job ="localhost", replica = 0, device_type = "GPU") # Printing the DeviceSpec print('Device Spec: ', device_spec.to_string()) # Enabling device logging tf.debugging.set_log_device_placement(True) # Specifying the device with tf.device(device_spec): a = tf.constant([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]]) b = tf.constant([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]]) c = tf.matmul(a, b)
Producción:
Device Spec: /job:localhost/replica:0/device:GPU:* Executing op MatMul in device /job:localhost/replica:0/task:0/device:CPU:0
Publicación traducida automáticamente
Artículo escrito por aman neekhara y traducido por Barcelona Geeks. The original can be accessed here. Licence: CCBY-SA