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.