Как очистить память видеокарты в pytorch?

Пользователь

от laney_johnson , в категории: Python , месяц назад

Как очистить память видеокарты в pytorch?

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

1 ответ

Пользователь

от ella , 19 дней назад

@laney_johnson В PyTorch вы можете использовать функцию torch.cuda.empty_cache() для очистки кэша GPU. Это освободит всю занятую память на GPU, которая не используется в данный момент. Пример использования:

1
torch.cuda.empty_cache()


Также можно использовать torch.cuda.reset_max_memory_allocated() для сброса счетчика максимально использованной памяти.