錯(cuò)誤原因:GPU資源占用太大
config = tf.ConfigProto(allow_soft_placement=True)
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)
config.gpu_options.allow_growth = True
找到如下紅的的這句話,在這之前加上如上三行代碼,在session前約束占用空間。能夠使得tensorflow占用資源降至70%,當(dāng)然也可以改為50%,而不是整個(gè)服務(wù)器GPU
with tf.Session() as sess:
將其改為
with tf.Session(config=config) as sess:
最終改為如下結(jié)構(gòu)就可以啦
config = tf.ConfigProto(allow_soft_placement=True)
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)
config.gpu_options.allow_growth = True
with tf.Session(config=config) as sess: