Gehen Sie mit der App Player FM offline!
DataLoader with multiple workers leaks memory
Manage episode 301257355 series 2921809
Today I'm going to talk about a famous issue in PyTorch, DataLoader with num_workers > 0 causes memory leak (https://github.com/pytorch/pytorch/issues/13246). This bug is a good opportunity to talk about DataSet/DataLoader design in PyTorch, fork and copy-on-write memory in Linux and Python reference counting; you have to know about all of these things to understand why this bug occurs, but once you do, it also explains why the workarounds help.
Further reading.
- A nice summary of the full issue https://github.com/pytorch/pytorch/issues/13246#issuecomment-905703662
- DataLoader architecture RFC https://github.com/pytorch/pytorch/issues/49440
- Cinder Python https://github.com/facebookincubator/cinder
83 Episoden
Manage episode 301257355 series 2921809
Today I'm going to talk about a famous issue in PyTorch, DataLoader with num_workers > 0 causes memory leak (https://github.com/pytorch/pytorch/issues/13246). This bug is a good opportunity to talk about DataSet/DataLoader design in PyTorch, fork and copy-on-write memory in Linux and Python reference counting; you have to know about all of these things to understand why this bug occurs, but once you do, it also explains why the workarounds help.
Further reading.
- A nice summary of the full issue https://github.com/pytorch/pytorch/issues/13246#issuecomment-905703662
- DataLoader architecture RFC https://github.com/pytorch/pytorch/issues/49440
- Cinder Python https://github.com/facebookincubator/cinder
83 Episoden
همه قسمت ها
×Willkommen auf Player FM!
Player FM scannt gerade das Web nach Podcasts mit hoher Qualität, die du genießen kannst. Es ist die beste Podcast-App und funktioniert auf Android, iPhone und im Web. Melde dich an, um Abos geräteübergreifend zu synchronisieren.