torch.cuda.memory.memory_reserved

torch.cuda.memory.memory_reserved(device=None)

返回给定设备上缓存分配器管理的当前 GPU 内存（以字节为单位）。

参数

device (torch.device or int, optional) – selected device. Returns statistic for the current device, given by current_device(), if device is None (default).

返回类型

int

注意

有关 GPU 内存管理的更多详细信息，请参阅 内存管理。