Session之配置选项

指定config参数,使用ConfigProto协议缓冲

config=tf.ConfigProto(log_device_placement=True)
#确保录入了运算分配到的GPU和CPU设备
config=tf.ConfigProto(allow_soft_placement=True)
#设备使用软约束
发布了46 篇原创文章 · 获赞 15 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/sinat_34686158/article/details/77662885