torch.cuda.set_per_process_memory_fraction

Определение

Метод PyTorch для ограничения доли общей GPU-памяти, доступной процессу. Позволяет симулировать работу с ограниченной памятью для тестирования.