fast-bert icon indicating copy to clipboard operation
fast-bert copied to clipboard
verified publisher icon utterworks

How do I clear GPU memory after this program finishes running

Open callmeYe opened this issue 5 years ago • 1 comments

I ran this program on Colab, and when I changed the parameters and tried to run it again, I got an error with CUDA out of memory. I can only kill the python3 process and rerun the entire notebook.

callmeYe avatar Jan 14 '21 02:01 callmeYe

torch.cuda.empty_cache()

lingdoc avatar Oct 20 '22 03:10 lingdoc