如何查找Kaggle笔记本的内存使用情况？

Question

我想知道如何查找Kaggle笔记本的内存使用情况。我如何才能发现运行Kaggle笔记本时已使用了多少内存。 Kaggle是允许python jupyter笔记本在其上运行的网站。

Answer 1

使用以下代码跟踪笔记本的内存使用情况：

import os, psutil  

def cpu_stats():
    pid = os.getpid()
    py = psutil.Process(pid)
    memory_use = py.memory_info()[0] / 2. ** 30
    return 'memory GB:' + str(np.round(memory_use, 2))