How can we release GPU memory cache?

28

Thank you Alexander the gc.collect() worked for me after deleting the model:

del model
gc.collect()
torch.cuda.empty_cache()

And I could check that memory was released using nvidia-smi