import tensorflow as tf
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
config.gpu_options.per_process_gpu_memory_fraction = 0.7
config.gpu_options.allocator_type = 'BFC'
关于BFC算法: 将内存分块管理,按块进行空间分配和释放。 通过split操作将大内存块分解成用户需要的小内存块。 通过merge操作合并小的内存块,做到内存碎片回收 通过bin这个抽象数据结构实现对空闲块高效管理。