Я недавно начал изучать Keras и TensorFlow. Я тестирую несколько моделей, которые в настоящее время находятся в наборе данных MNIST (довольно базовый материал c). Я хотел знать, сколько именно моя модель потребляет в памяти во время тренировок и умозаключений. Я попробовал поискать в Google, но не нашел много информации.
Я наткнулся на Nvidia-smi. Я попытался использовать config.gpu_options.allow_growth = True , но все еще не могу использовать точную память, которую python .exe использует из-за некоторых проблем с Nvidia-smi. Я знаю, что мог бы запустить отдельный проход поезда и вывода, но это слишком громоздко. Это было бы очень просто, если бы я мог просто найти подходящий API для этой работы.
Tensorflow, будучи такой хорошо известной и широко используемой библиотекой, я надеюсь найти лучший и более быстрый способ добраться до этих чисел ,
Наконец, еще раз, мой вопрос: как получить точное использование памяти для модели Keras во время обучения и логического вывода.
Соответствующие спецификации: ОС: Windows 10 GPU: GTX 1050 TensorFlow версия: 1.14
Пожалуйста, дайте мне знать, если требуются какие-либо другие детали.
Спасибо!