Неограниченное потребление оперативной памяти cnn, lasagne, ubuntu 16.04

[FilippNikitin]

Сверточная нейронная сеть с парой слоев (элементарный пример) при обучении начинает линейно увеличивать потребление оперативной памяти (вплоть до 16гб + 10 гб файла подкачки). Слышал от ребят, что такая проблема встречалась, может кто знает, в чем дело?
Стоит ubuntu 16.04 (под windows данной проблемы нет).
Anaconda 2.7 64 bit
Последние версии lasagne и theano.

[Nenteru]

У меня немного другая проблема.

Linux Ubuntu 22.04 LTS
Anaconda 24.7.1

Тренировал нейронку на 2 ляма параметров, видюху грузила на 5 гб и оперативы тоже 5 гб в пике, однако остальные 50 с лишним гб оперативы грузились кэшем, что приводило к ошибке: [tensorflow/stream_executor/cuda/cuda_fft.cc:439] failed to initialize batched cufft plan with customized allocator: Failed to make cuFFT batched plan. Aborted (core dumped).

Сейчас как раз пытаюсь решить эту проблему, возможно не верно генерирую датасет, либо дырки в ПО tensorflow версии 2.9.