Ограничение использования памяти графическим процессором Keras / TF 2019? - PullRequest
0 голосов
/ 22 апреля 2019

Я прочитал ответы типа:

import tensorflow as tf
from keras.backend.tensorflow_backend import set_session
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.2
set_session(tf.Session(config=config))

Но это просто не работает. Кажется, что в керасе и TF так много обновлений, что почти все, что написано в 2017 году, не работает! Итак, как ограничить использование памяти?

...