在每个训练时期后获取内存使用情况 - Tensorflow 1.15 版

Question

我正在使用需要 tensorflow 版本 1.15 的代码。

但是，我想记录 BERT 模型的每个训练时期后使用的内存量。在 tensorflow 版本 2 中，这可以通过 tf.config.experimental.get_memory_usage 实现。我怎样才能对 1.15 版做同样的事情？如果它不是 tensorflow 的功能，是否有另一种方法可以记录每个纪元后的内存使用情况？

提前致谢，如果问题很简单 - 初学者，我们深表歉意

Answer 1

定期调用 psutil 向操作系统询问最近的资源消耗情况。

import psutil

mem = psutil.virtual_memory()
print(mem)