Pytorch: How to clear GPU memory

import gc

# del optimizer

# del model

gc.collect()

torch.cuda.empty_cache()