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

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

图像加载尺寸和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比特数显式定义了没?
页: [1]
查看完整版本: 同样参数下3090和A6000炼丹占用显存不同