Esta página foi traduzida pela API Cloud Translation.
Switch to English

tf.device

Versão TensorFlow 1 Ver fonte no GitHub

Especifica o dispositivo para operações criadas / executadas neste contexto.

Usado nos notebooks

Utilizado no guia Usado nos tutoriais

Esta função especifica o dispositivo a ser usado para operações criadas / executadas em um contexto específico. Os contextos aninhados herdarão e também criarão / executarão suas operações no dispositivo especificado. Se um dispositivo específico não for necessário, considere não usar esta função para que um dispositivo possa ser atribuído automaticamente. Em geral, o uso desta função é opcional. device_name pode ser totalmente especificado, como em "/ job: worker / task: 1 / device: cpu: 0" ou parcialmente especificado, contendo apenas um subconjunto dos campos separados por "/". Quaisquer campos especificados substituirão as anotações do dispositivo dos escopos externos.

Por exemplo:

 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 O nome do dispositivo a ser usado no contexto.

Um gerenciador de contexto que especifica o dispositivo padrão a ser usado nas operações recém-criadas.

RuntimeError Se uma função é passada.