Halaman ini diterjemahkan oleh Cloud Translation API.
Switch to English

tf.config.set_logical_device_configuration

Lihat sumber di GitHub

Mengatur konfigurasi perangkat logis untuk tf.config.PhysicalDevice .

Digunakan di notebook

Digunakan dalam panduan

Sebuah terlihat tf.config.PhysicalDevice akan secara default memiliki satu tf.config.LogicalDevice terkait dengan sekali runtime diinisialisasi. Menentukan daftar tf.config.LogicalDeviceConfiguration objek memungkinkan beberapa perangkat yang akan dibuat pada yang sama tf.config.PhysicalDevice .

Contoh berikut membagi CPU menjadi 2 perangkat logis:

physical_devices = tf.config.list_physical_devices('CPU')
assert len(physical_devices) == 1, "No CPUs found"
# Specify 2 virtual CPUs. Note currently memory limit is not supported.
try:
  tf.config.set_logical_device_configuration(
    physical_devices[0],
    [tf.config.LogicalDeviceConfiguration(),
     tf.config.LogicalDeviceConfiguration()])
  logical_devices = tf.config.list_logical_devices('CPU')
  assert len(logical_devices) == 2

  tf.config.set_logical_device_configuration(
    physical_devices[0],
    [tf.config.LogicalDeviceConfiguration(),
     tf.config.LogicalDeviceConfiguration(),
     tf.config.LogicalDeviceConfiguration(),
     tf.config.LogicalDeviceConfiguration()])
except:
  # Cannot modify logical devices once initialized.
  pass

Contoh berikut membagi GPU menjadi 2 perangkat logis dengan masing-masing 100 MB:

physical_devices = tf.config.list_physical_devices('GPU')
try:
  tf.config.set_logical_device_configuration(
    physical_devices[0],
    [tf.config.LogicalDeviceConfiguration(memory_limit=100),
     tf.config.LogicalDeviceConfiguration(memory_limit=100)])

  logical_devices = tf.config.list_logical_devices('GPU')
  assert len(logical_devices) == len(physical_devices) + 1

  tf.config.set_logical_device_configuration(
    physical_devices[0],
    [tf.config.LogicalDeviceConfiguration(memory_limit=10),
     tf.config.LogicalDeviceConfiguration(memory_limit=10)])
except:
  # Invalid device or cannot modify logical devices once initialized.
  pass

device The PhysicalDevice untuk mengkonfigurasi.
logical_devices (opsional) Daftar tf.config.LogicalDeviceConfiguration objek untuk mengalokasikan untuk ditentukan PhysicalDevice . Jika ada, konfigurasi default akan digunakan.

ValueError Jika argumen validasi gagal.
RuntimeError Runtime sudah diinisialisasi.