Se usó la API de Cloud Translation para traducir esta página.
Switch to English

tf.device

Versión TensorFlow 1 Ver código fuente en GitHub

Especifica el dispositivo para operaciones creadas / ejecutadas en este contexto.

Usado en los cuadernos

Usado en la guía Usado en los tutoriales

Esta función especifica el dispositivo que se utilizará para las operaciones creadas / ejecutadas en un contexto particular. Los contextos anidados heredarán y también crearán / ejecutarán sus operaciones en el dispositivo especificado. Si no se requiere un dispositivo específico, considere no usar esta función para poder asignar un dispositivo automáticamente. En general, el uso de esta función es opcional. device_name se puede especificar completamente, como en "/ trabajo: trabajador / tarea: 1 / dispositivo: cpu: 0", o parcialmente especificado, que contiene solo un subconjunto de los campos separados por "/". Cualquier campo que se especifique anulará las anotaciones del dispositivo de los ámbitos externos.

Por ejemplo:

 with tf.device('/job:foo'):
  # ops created here have devices with /job:foo
  with tf.device('/job:bar/task:0/device:gpu:2'):
    # ops created here have the fully specified device above
  with tf.device('/device:gpu:1'):
    # ops created here have the device '/job:foo/device:gpu:1'
 

device_name El nombre del dispositivo a usar en el contexto.

Un administrador de contexto que especifica el dispositivo predeterminado para usar para operaciones recién creadas.

RuntimeError Si se pasa una función.