PyTorch Forums
Model.to("cpu") does not release GPU memory allocated by registered buffer
ljj7975
(Brandon Lee)
July 8, 2021, 6:53pm
6
“you cannot delete the CUDA context while the PyTorch process is still running”
1 Like
show post in topic