tensorflow中使用tf.ConfigProto()配置Session运行参数&&GPU设备指定
原文:https://blog.csdn.net/dcrmg/article/details/79091941
tf.ConfigProto()函数用在创建session的时候,用来对session进行参数配置:
config = tf.ConfigProto(allow_soft_placement=True, allow_soft_placement=True)
config.gpu_opti...
转载
2019-02-25 19:00:55 ·
408 阅读 ·
0 评论