torch.cuda.memory_usage

torch.cuda.memory_usage(device=None)

返回在过去采样周期内，全局（设备）内存被读取或写入的时间百分比，如nvidia-smi所示。

Parameters

设备 (torch.device 或 int, 可选) – 选择的设备。返回当前设备的统计信息，由 current_device() 给出，如果 device 为 None（默认）。

Return type

int

警告：每个采样周期可能介于1秒到1/6秒之间，具体取决于所查询的产品。