文件名称：gpu_memory_profiling:剖析Pytorch代码中每一行的GPU内存使用情况
文件大小：4KB
文件格式：ZIP
更新时间：2024-06-02 08:24:37
Python
gpu_memory_profiling 在pytorch代码中分析每一行的GPU内存使用情况 用法示例 python example_mnist.py 相依性 此代码取决于 。 点安装在这里可用： pip install py3nvml 使用pytorch版本0.4.0和py3nvml版本0.2.0进行了测试 致谢 gpu_profile.py是的修改版本。 从此修改了example_mnist.py。
【文件预览】：
gpu_memory_profiling-master
----README.md(604B)
----example_mnist.py(5KB)
----.gitignore(27B)
----gpu_profile.py(4KB)