Stonesjtu / pytorch_memlab

Profiling and inspecting memory in pytorch
MIT License
1.01k stars 37 forks source link

Clearing GPU/CUDA memory #21

Closed Buckeyes2019 closed 1 year ago

Buckeyes2019 commented 3 years ago

This is a very interesting library and will hopefully help with constant OOM errors. Beyond profiling the memory usage, does memlab offer a way to clear GPU memory (if we do find the GPU holding onto tensors after execution)? Thanks!