Halaman ini diterjemahkan oleh Cloud Translation API.
Switch to English

tf.config.set_visible_devices

Lihat sumber di GitHub

Mengatur daftar perangkat yang terlihat.

Digunakan di notebook

Digunakan dalam panduan

Menentukan PhysicalDevice benda yang terlihat runtime. TensorFlow hanya akan mengalokasikan operasi memori dan tempat pada perangkat fisik yang terlihat, karena kalau tidak ada LogicalDevice akan dibuat pada mereka. Secara default semua perangkat ditemukan ditandai sebagai terlihat.

Contoh berikut menunjukkan menonaktifkan GPU pertama pada mesin.

physical_devices = tf.config.list_physical_devices('GPU')
try:
  # Disable first GPU
  tf.config.set_visible_devices(physical_devices[1:], 'GPU')
  logical_devices = tf.config.list_logical_devices('GPU')
  # Logical device was not created for first GPU
  assert len(logical_devices) == len(physical_devices) - 1
except:
  # Invalid device or cannot modify virtual devices once initialized.
  pass

devices Daftar PhysicalDevice s untuk membuat terlihat
device_type (Opsional) perangkat mengkonfigurasi Hanya cocok jenis perangkat ini. Misalnya "CPU" atau "GPU". Perangkat lain akan ditinggalkan tidak berubah.

ValueError Jika argumen validasi gagal.
RuntimeError Runtime sudah diinisialisasi.