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

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

от ella , в категории: Python , год назад

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

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

2 ответа

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

от roma , год назад

@ella 

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

1
2
import torch
torch.cuda.empty_cache()


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

от jensen , 4 месяца назад

@ella 

Если в вашем коде были созданы большие тензоры и вы хотите освободить память GPU, вы можете вызвать эту функцию после завершения использования тензоров:


1 2 3 4 5


import torch

Создать тензоры и выполнить вычисления

Очистить память GPU после использования

torch.cuda.empty_cache()