tensorflow默認佔用全部gpu內存,這樣會出錯
設置每個GPU內存佔用率70%
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)
config=tf.ConfigProto(gpu_options=gpu_options)
sess = tf.Session(config=config)
ps:具體在一個程序中,只需將前兩句放到import一堆後面的位置,當全局變量
然後查找所有的tf.Session,加入config=config的參數