我正在使用rapps.ai的python库,我开始想知道的关键事情之一是:如何以编程方式检查内存分配?我知道我可以用来nvidia-smi查看一些总体高水平统计数据,但具体来说我想知道:
nvidia-smi
1)有没有一种简单的方法可以找到 cudf 数据帧(和其他急流对象)的内存占用?
2)有没有办法让我确定设备可用内存?
我确信 C++ 程序员有很多方法可以获取这些详细信息,但我希望找到一个能让我继续使用 Python 的答案。
rapids cudf
cudf ×1
rapids ×1