torch.cuda.memory_reserved

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

Возвращает текущий объём памяти GPU, управляемой кэширующим выделением памяти, в байтах для заданного устройства.

Параметры

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

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

int

Примечание

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