Rate this Page
★★★★★

torch.cuda.memory.mem_get_info#

torch.cuda.memory.mem_get_info(device=None)[source]#

Return the global free and total GPU memory for a given device using cudaMemGetInfo.

Parameters

device (torch.device orint orstr,optional) – selected device. Returnsstatistic for the current device, given bycurrent_device(),ifdevice isNone (default) or if the device index is not specified.

Return type

tuple[int,int]

Note

SeeMemory management for moredetails about GPU memory management.