Как очистить память gpu в pytorch?

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

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

Как очистить память gpu в pytorch?

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

1 ответ

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

от roma , 5 месяцев назад

@ella 

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

1
2
import torch
torch.cuda.empty_cache()