萌新求问再用tensorflow-gpu时发现显存不够,百度
tensorflow吧
全部回复
仅看楼主
level 2
0Kº 楼主
萌新求问
再用tensorflow-gpu时发现显存不够,百度出来设置
config=tf.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.Session(config=config)
这是每一次使用都必须要写的吗?
不能再外部设置一次,以后默认开启allow_growth
用的tensorflow-gpu 1.12
2019年10月01日 05点10分 1
1