torch.accelerator.memory.empty_cache

torch.accelerator.memory.empty_cache() [source]

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

Примечание

Эта функция ничего не делает, если аллокатор памяти для текущего ускорителя не был инициализирован.