Halaman ini diterjemahkan oleh Cloud Translation API.
Switch to English

tf.device

TensorFlow 1 versi Lihat sumber di GitHub

Menentukan perangkat untuk ops dibuat / dilaksanakan dalam konteks ini.

Digunakan di notebook

Digunakan dalam panduan Digunakan dalam tutorial

Fungsi ini menentukan perangkat yang akan digunakan untuk ops dibuat / dilaksanakan dalam konteks tertentu. konteks bersarang akan mewarisi dan juga membuat / mengeksekusi ops mereka pada perangkat tertentu. Jika perangkat tertentu tidak diperlukan, pertimbangkan untuk tidak menggunakan fungsi ini sehingga perangkat dapat secara otomatis ditetapkan. Secara umum penggunaan fungsi ini adalah opsional. device_name dapat sepenuhnya ditentukan, seperti dalam "/ Pekerjaan: Pekerja / tugas: 1 / perangkat: cpu: 0", atau sebagian ditentukan, hanya berisi subset dari "/" - kolom yang dipisahkan. Setiap bidang yang ditentukan akan menimpa penjelasan perangkat dari lingkup luar.

Sebagai contoh:

 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 Nama perangkat untuk digunakan dalam konteks.

Seorang manajer konteks yang menentukan perangkat default untuk digunakan untuk ops baru dibuat.

RuntimeError Jika fungsi dilewatkan di.