【tensorflow】设置显存开销

问题

一般大家在跑tf时，单个程序往往会占满整块GPU的所有显存。 但是实际上，程序很可能并不需要那么大的显存空间。

改进方案

通过 tf.ConfigProto().gpu_options.allow_growth=True来告诉程序，只需占用实际所需的显存即可：

代码语言：javascript复制

# ---------------- session ----------------
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.Session(config=config)

如果这里把 config.gpu_options.allow_growth设置为False，那么程序就会默认把整块卡的所有显存占满。

config gpu 程序

0 人点赞