Player FMアプリでオフラインにしPlayer FMう!
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 つのエピソード
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 つのエピソード
すべてのエピソード
×プレーヤーFMへようこそ!
Player FMは今からすぐに楽しめるために高品質のポッドキャストをウェブでスキャンしています。 これは最高のポッドキャストアプリで、Android、iPhone、そしてWebで動作します。 全ての端末で購読を同期するためにサインアップしてください。