方式如下：
gpu_options = tf.GPUOptions(allow_growth=True)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
方式如下：
gpu_options = tf.GPUOptions(allow_growth=True)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
本文标题：TensorFlow 按需使用显存设置
本文链接：https://www.haomeiwen.com/subject/pframctx.html
网友评论