Как помешать тензорному потоку выделить всю память GPU?

Я работаю в среде, в которой вычислительные ресурсы используются совместно, то есть у нас есть несколько серверных машин, оснащенных несколькими графическими процессорами Nvidia Titan X на каждой. Для моделей малого и среднего размера 12 ГБ Titan X обычно достаточно, чтобы 2-3 человека могли...