使用GPU運算的時候出現TensorFlow CUDA_ERROR_OUT_OF_MEMORY

在linux下運行fcn出現failed to allocate錯誤，可以從如下幾個方面提高GPU的利用率：

1. reduce your batch size

2 .use memory growing:

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config, ...)

3 .don't allocate whole of your GPU memory(only 90%):

config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.9
session = tf.Session(config=config, ...)

 

 通過以上設定就可以解決GPU分配的問題了，也可以通過以上設定運行程序的GPU的最大佔用率。