Пользователь
ella
Как очистить память gpu в pytorch?
roma
@ella
В PyTorch можно очистить память GPU, используя функцию torch.cuda.empty_cache(). Эта функция сбрасывает кэш GPU, удаляя все неиспользуемые тензоры. Например:
1 2
import torch torch.cuda.empty_cache()