有没有办法打印出Python程序运行时的GPU内存使用情况？

Question

我正在尝试优化我的 python 程序的 GPU 内存使用情况，在任务管理器上我可以看到它在一段时间内保持较低水平，然后在某个时刻它会上升。我不确定这是在程序的哪个点上，所以我想知道是否有办法在程序运行时将其打印到屏幕上，以帮助我找到突然增加的原因？正如您从屏幕截图中看到的，我的内存和 CPU 使用率也几乎同时上升，因此即使前者不可能，任何有关如何打印这些内容的评论将不胜感激。

Answer 1

看起来你是在 Windows 上，这样做更具挑战性。在 Linux 上，您只需在代码中添加一些 !nvidia-smi 命令，它就会读出 GPU 使用信息。可以对 Linux 上的 CPU 和内存使用情况执行类似的命令。