slim是Tensorflow于2016推出的模块,是一种high-level库,讲许多深度学习算法封装起来,极大地简化了Tensorflow的代码。对于一般的神经网络训练,slim封装了slim.learning.train方法,连Session都不需要创建了。这样,GPU显存配置项也发生变化:
方式1,按需分配:
session_config = tf.ConfigProto()
session_config.gpu_options.allow_growth = True
slim.learning.train(..., session_config=session_config)
方式2,自定义比例;
session_config = tf.ConfigProto()
session_config.gpu_options.per_process_gpu_memory_fraction = 0.45
slim.learning.train(..., session_config=session_config)
如果不设置,默认为全占。