torch.cuda.memory.get_per_process_memory_fraction

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

Получить долю памяти для процесса.

Параметры:

device (torch.device or int, необязательно) – выбранное устройство. Если это None, используется устройство CUDA по умолчанию.

Возвращает:

доля памяти в диапазоне 0~1. Допустимый объём памяти равен total_memory * fraction.

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

float