pytorch测试内存泄漏

最新推荐文章于 2024-07-25 14:52:49 发布

追梦小狂魔

最新推荐文章于 2024-07-25 14:52:49 发布

阅读量190

点赞数 1

文章标签： pytorch 人工智能 python

本文链接：https://blog.youkuaiyun.com/qq_15557299/article/details/135858448

版权

文章介绍了如何使用Python的psutil库来监控当前进程的内存使用情况，通过循环和迭代，每10次迭代显示一次内存使用量（以MB为单位）。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import psutil
    process = psutil.Process(os.getpid())
    for _ in range(10):
        for i, batch in tqdm(enumerate(train_data)):
            img, text = batch
            mm_info = process.memory_full_info()
            print(i, mm_info.uss / 1024 / 1024, "MB")