torch.cuda.memory.empty_cache

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

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

Примечание

empty_cache() не увеличивает объём памяти GPU, доступной PyTorch. Однако в некоторых случаях это может помочь уменьшить фрагментацию памяти GPU. Подробнее об управлении памятью GPU см. в разделе Управление памятью.