Data ProfilingOpen Source Data Platforms

Pytorch Memlab

Profiling and inspecting memory in pytorch

Open Source

About

Out-Of-Memory errors in pytorch happen frequently, for new-bees and experienced programmers. A common reason is that most people don't really learn the underlying memory management philosophy of pytorch and GPUs. They wrote memory in-efficient codes and complained about pytorch eating too much CUDA memory.

Open Source Health

Not enough history
Stars
1,078
Forks
39
License
MIT
Last commit
2 months ago
Python

Related Categories