torch.cuda.memory_usage

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

Возвращает процент времени за последний период выборки, в течение которого глобальная (устройства) память читалась или записывалась, как указано в nvidia-smi.

Parameters:

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

Return type:

int

Предупреждение: Каждый период выборки может составлять от 1 секунды до 1/6 секунды, в зависимости от запрашиваемого продукта.