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

gx19860411 · 发表于 2022-1-3 21:38

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

迷之路人 · 发表于 2022-1-4 03:07

你这两个卡的服务器用的环境是一样的么？如果不一样，可以用Nvidia的docker container 试一下。tensorflow，pytorch

Gazzz · 发表于 2022-1-4 09:13

本帖最后由 Gazzz 于 2022-1-4 09:14 编辑

检查下环境是否一致有没有apex torch/tf环境还有代码本身都有可能影响显存

gx19860411 · 发表于 2022-1-4 11:45

好的我查一下。
话说炼丹问题该去哪问人比较多呢？潭里的炼丹群基本死寂了

—— 来自 Xiaomi M2011K2C, Android 11上的 S1Next-鹅版 v2.5.2

twy_2000 · 发表于 2022-1-4 12:31

torch环境，驱动版本，cuda版本。

我老婆呢 · 发表于 2022-1-5 01:47

float比特数显式定义了没？来自: iPhone客户端

		自动登录	找回密码
密码			立即注册

[软件] 同样参数下3090和A6000炼丹占用显存不同