本文介绍如何在TensorFlow程序中限制GPU显存使用，通过配置会话参数，可以指定GPU显存使用的具体比例，避免程序占用全部显存资源。
一般情况下，tf程序运行时会占用gpu所有显存。可以通过会话设置，使得代码只是用gpu一定比例的显存：
config = tf.ConfigProto()
congfig.gpu_options.per_process_gpu_memory_fraction = 0.4
sess = tf.Session(config =config )
Tensorflow设置gpu显存使用比例
一般情况下，tf程序运行时会占用gpu所有显存。可以通过会话设置，使得代码只是用gpu一定比例的显存：
config = tf.ConfigProto()
congfig.gpu_options.per_process_gpu_memory_fraction = 0.4
sess = tf.Session(config =config )
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
扫一扫
