按需要自动分配，避免Tensorflow一次性占满
gpu_options = tf.GPUOptions(allow_growth=True) sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options)) # sess = tf.InteractiveSession(config=tf.ConfigProto(gpu_options=gpu_options))文章来源: Tensorflow设置GPU显存按需分配