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

@ella 

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

1
2

import torch
torch.cuda.empty_cache()