The memory does not free up after running the code once. Is there way to free up the memory without restarting the kernel?
You could try adding:torch.cuda.empty_cache()
torch.cuda.empty_cache()
If that doesn't work then I'm not sure
encountering the same problem.
· Sign up or log in to comment
