我最近开始学习Keras和TensorFlow。我正在测试MNIST数据集上的一些模型（相当基本的东西）。我想知道，在训练和推理期间，我的模型究竟在内存上消耗了多少。我尝试使用Google搜索，但没有找到太多信息。
我遇到了Nvidia-smi。我尝试使用 config.gpu_options.allow_growth = True 选项，但由于Nvidia-smi的某些问题，仍然无法使用python.exe正在消耗的确切内存。我知道我可以单独进行训练和推理，但这太麻烦了。如果我能找到合适的API来完成这项工作，这将非常容易。
Tensorflow是一个如此知名且使用广泛的库，我希望找到一种更好，更快的方法来获取这些数字。
最后，我的问题再次是： 如何在训练和推理过程中获得Keras模型的确切内存使用情况。
相关规格： 作业系统：Windows 10 GPU：GTX 1050 TensorFlow版本：1.14
如果需要其他详细信息，请告诉我。
谢谢！