同样参数下3090和A6000炼丹占用显存不同
图像加载尺寸和batchsize啥的全都一样,自己的3090占用22g显存,但租的服务器是A6000显卡,要占用35g的显存,不知道怎么回事,求指点 你这两个卡的服务器用的环境是一样的么?如果不一样,可以用Nvidia的docker container 试一下。tensorflow,pytorch 本帖最后由 Gazzz 于 2022-1-4 09:14 编辑检查下环境是否一致 有没有apex torch/tf环境 还有代码本身都有可能影响显存 好的我查一下。
话说炼丹问题该去哪问人比较多呢?潭里的炼丹群基本死寂了
—— 来自 Xiaomi M2011K2C, Android 11上的 S1Next-鹅版 v2.5.2 torch环境,驱动版本,cuda版本。 float比特数显式定义了没?
页:
[1]