torch.cuda.memory.mem_get_info

torch.cuda.memory.mem_get_info(device=None) [исходный код]

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

Параметры:

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

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

tuple[int, int]

Примечание

Подробнее об управлении памятью GPU см. в разделе Управление памятью.