pytorch - 训练 DNN 时如何监控 GPU 内存使用情况？

Question

我举一个结果例子。我想问如何获得像这张图这样的数据。

标签： pytorchgpu

Answer 1

您可以使用 pytorch 命令，例如torch.cuda.memory_stats获取有关当前 GPU 内存使用情况的信息，然后根据这些报告创建时间图。