TensorFlow限制占用显存比例以及实现占用显存自增加

当实验室的服务器紧张,一张卡要几我的用时,就得预留显存给别人了,若是默认模式,TensorFlow会占用全部的freememory,这就不友好了。除此以外,若是指定显存自增加,也方便你知道本身的程序须要多大的显存。web tf_config = tf.ConfigProto() tf_config.gpu_options.per_process_gpu_memory_fraction = 0.
相关文章
相关标签/搜索