pytorch_memlab:在pytorch中分析和检查内存

pytorch_memlab 一个简单，准确的pytorch CUDA内存管理实验室，它由有关内存的不同部分组成： 特征： 内存探查器：具有简单API的line_profiler样式CUDA内存探查器。 内存报告器：一个报告器，用于检查占用CUDA存储器的张量。 礼貌：一个有趣的功能，可以将所有CUDA张量临时移到CPU内存中以礼貌，当然也可以向后传送。 IPython通过%mlrun / %%mlrun行/单元魔术命令提供支持。 目录 安装 发布版本： pip install pytorch_memlab 最新版本： pip install git+https://github.com/stonesjtu/pytorch_memlab 有什么用 对于新手和有经验的程序员，pytorch中的内存不足错误经常发生。 一个普遍的原因是，大多数人并没有真正了解pytorch和