内存释放问题

[mmlingyu]

/root/anaconda3/envs/muse/bin/python 7624MiB
启动运行过后 gpu一直在7g左右，请问哪里可以优化释放内存. 里面unet 和 vae 用的么？

[zhangziliang04]

推理过程中，回加载模型，模型大小bs=4的时候，大概7G，推理结束会自动释放。

[mmlingyu]

现在是处理完内存还是没有释放 是哪一个地方释放的 我仔细看一下代码  发自我的iPhone

…

------------------ 原始邮件 ------------------ 发件人: 张子良 ***@***.***> 发送时间: 2024年8月17日 19:39 收件人: TMElyralab/MuseTalk ***@***.***> 抄送: mmlingyu ***@***.***>, Author ***@***.***> 主题: Re: [TMElyralab/MuseTalk] 内存释放问题 (Issue #168) 推理过程中，回加载模型，模型大小bs=4的时候，大概7G，推理结束会自动释放。 — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: ***@***.***>