Halaman ini diterjemahkan oleh Cloud Translation API.
Switch to English

tf.config.experimental.set_memory_growth

TensorFlow 1 versi Lihat sumber di GitHub

Mengatur jika pertumbuhan memori harus diaktifkan untuk PhysicalDevice .

Digunakan di notebook

Digunakan dalam panduan Digunakan dalam tutorial

Jika pertumbuhan memori diaktifkan untuk PhysicalDevice , runtime inisialisasi tidak akan mengalokasikan semua memori pada perangkat. Pertumbuhan memori tidak dapat dikonfigurasi pada PhysicalDevice dengan perangkat virtual dikonfigurasi.

Sebagai contoh:

physical_devices = tf.config.list_physical_devices('GPU')
try:
  tf.config.experimental.set_memory_growth(physical_devices[0], True)
except:
  # Invalid device or cannot modify virtual devices once initialized.
  pass

device PhysicalDevice untuk mengkonfigurasi
enable (Boolean) Apakah untuk mengaktifkan atau menonaktifkan pertumbuhan memori

ValueError Valid PhysicalDevice ditentukan.
RuntimeError Runtime sudah diinisialisasi.