torch.cuda.empty_cache

torch.cuda.empty_cache() [source]

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

Примечание

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