torch.cuda.mem_get_info

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

Возвращает объём глобальной свободной и общей памяти GPU для заданного устройства, используя cudaMemGetInfo.

Параметры:

device (torch.device или int, необязательно) – выбранное устройство. Возвращает статистику для текущего устройства, заданного с помощью current_device(), если device равен None (по умолчанию).

Тип возвращаемого значения:

Tuple[int, int]

Примечание

См. Управление памятью для получения дополнительной информации об управлении памятью GPU.