Как установить динамический рост памяти c на TF 2.1? - PullRequest
0 голосов
/ 04 февраля 2020

В предыдущих версиях tenorflow + keras я мог установить опцию allow_growth и просматривать использование памяти в реальном времени с помощью nvidia-smi. В противном случае все это будет немедленно выделено процессом. Теперь, используя tf.keras в tenorflow 2.1, я не могу найти способ сделать это. Любая помощь приветствуется!

1 Ответ

1 голос
/ 04 февраля 2020

Добавление этого в начало моей программы дало мне желаемое поведение

physical_devices = tf.config.list_physical_devices('GPU') 
tf.config.experimental.set_memory_growth(physical_devices[0], True)
...