[TensorFlow] Allocator (GPU_0_bfc) ran out of memory trying to allocate

 

 

 

 

 

Allocator (GPU_0_bfc) ran out of memory trying to allocate 에러는 아래와 같이 gpu 옵션을 통해 해결

 

 

방법 1 : gpu 메모리 런타임 할당에 따라 메모리 설정하여 해결 ( allow_growth 이용)

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

 

방법 2 : gpu 메모리 할당 비율을 설정하여 해결 (per_process_gpu_memory_fraction 이용)

config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.4
session = tf.Session(config=config)

 

 

참고자료 : 

https://stackoverflow.com/questions/36927607/how-can-i-solve-ran-out-of-gpu-memory-in-tensorflow

How can I solve 'ran out of gpu memory' in TensorFlow