【文件属性】：
文件名称：pytorch_memlab:在pytorch中分析和检查内存
文件大小：90KB
文件格式：ZIP
更新时间：2021-05-04 18:31:52
pytorch memory-profiler cuda-memory Python
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和
【文件预览】：
pytorch_memlab-master
----setup.py(1010B)
----.gitignore(2KB)
----requirements.txt(54B)
----.travis.yml(1KB)
----LICENSE(1KB)
----demo.ipynb(32KB)
----LICENSE_kernprof.txt(2KB)
----README.md(17KB)
----readme-output.png(72KB)
----pytorch_memlab()
--------mem_reporter.py(7KB)
--------__init__.py(275B)
--------courtesy.py(1KB)
--------utils.py(180B)
--------line_profiler()
----test()
--------test_courtesy.py(759B)
--------__init__.py(0B)
--------test_line_profiler.py(3KB)
--------test_mem_reporter.py(2KB)