同一个模型不同版本的软件使用显存差距为什么这么大

fghfdg

星级打分

• 1
• 2
• 3
• 4
• 5

平均分:NAN  参与人数:0  我的评分:未评

模型是自己预训练的liae-udt320-448-112-112-34，bs都是4，训练的参数也一样



这是在dml版本的软件上面的显存使用情况




这是在dx12l版本的软件上面的显存使用情况

各位N卡的大佬在训练时会存在这个问题吗

654550006

你第一张图不准，前面半截基本满载了，你后面应该停止GPU了

fghfdg

654550006 发表于 2025-3-31 09:26
你第一张图不准，前面半截基本满载了，你后面应该停止GPU了

这个不影响的，截图的时候我把模型停了，刚好是那个瞬间。当模型完整保存以后显存占用才会变小的，不然会一直维持训练时候的占用情况（bs4占用21g）我的问题就是同一个模型在dml和dx12上训练显存差距为什么这么大