How to free GPU memory

[boatingMen]

When I write the text video retrieval function as a Python script, when using the function, GPU video memory increases with the increase of the number of uses (search), and the kill script is released, I want to ask whether the existing method can release video memory?

[riichg5]

you can use torch.cuda.empty_cache() in you code at any where.

import torch
torch.cuda.empty_cache()