torch.mps.empty_cache

torch.mps.empty_cache() [исходный код]

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