ResourceExhaustedError: OOM when allocating tensor with shape[1,128,256,256] and type float ……

運行代碼時，報標題錯。

原因顯存不夠：

可以從以下幾處排查：

1、tensor shape[1,128,256,256]，batch_size可以改小，我這裏是1了，後面256x256圖片尺寸也可以改小，

2、我改了訓練集batch後發現還是報錯，最後發現是測試集幾次性輸入所有圖片，所以要注意排查測試集和數據集都要分批送進網絡

3、在之前改bug的時候- -幹了個蠢事：

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.1)
with tf.Session(config = tf.ConfigProto(gpu_options=gpu_options)) as sess:

有時候tensorflow自己會自動退出，是因爲tf默認佔所有顯存，如果有別的進程在使用顯存，可能會自動退出。所以加了上面這句話，但指定中0.1的顯存這也太少了- -，改了之後就好了。

總之！不要瞎改bug！！！！！切記！