TensorFlow限制佔用顯存比例以及實現佔用顯存自增加

當實驗室的服務器緊張，一張卡要幾我的用時，就得預留顯存給別人了，若是默認模式，TensorFlow會佔用全部的freememory，這就不友好了。除此以外，若是指定顯存自增加，也方便你知道本身的程序須要多大的顯存。web tf_config = tf.ConfigProto() tf_config.gpu_options.per_process_gpu_memory_fraction = 0.

>>阅读原文<<