Haoxiang’s Blog
Skip to content
  • Home
  • Blog Posts
  • Professional Profile

pytorch

There is one post tagged pytorch.

A PyTorch GPU Memory Leak Example

I ran into this GPU memory leak issue when building a PyTorch training pipeline. After spending quite some time, I finally figured out this minimal reproducible example. Kicking off the training, it shows constantly increasing allocated GPU memory. This “AverageMeter” has been used in many popular repositories (e.g., https://github.com/facebookresearch/moco). It’s by-design tracking the average of […]

in Blog Posts, Solution | 611 Words | Comment
这个网站使用的主机 (我的推广链接)
Independent Publisher empowered by WordPress