TensorFlow 按需使用显存设置

方式如下:

gpu_options = tf.GPUOptions(allow_growth=True)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

你可能感兴趣的:(TensorFlow 按需使用显存设置)