torch.cuda.empty_cache

torch.cuda.empty_cache() [source]

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

Примечание

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