同样参数下3090和A6000炼丹占用显存不同

THCEnthusiast

2022-01-02T17:00:51+00:00

图像加载尺寸和batchsize啥的全都一样，自己的3090占用22g显存，但租的服务器是A6000显卡，要占用35g的显存，不知道怎么回事，求指点

Tj🍜

这俩驱动都不一样吧

RiverDragonz

a6000是sm80 3090是sm86 不过一般论区别不在这里多半是malloc看你剩余多就多要了点。。。

torch.backends.cudnn.deterministic之类的还有些别的自己开一下，但是会变慢很多。

Mano Bro

为了加快训练会多占用一些，这个是框架里定义的策略，可以改。之前的keras默认吃满显存。

THCEnthusiast

明白了，多谢楼上各位