首页 > 其他 > 详细

【转载】 tf.ConfigProto和tf.GPUOptions用法总结

时间:2019-12-13 14:31:02      阅读:116      评论:0      收藏:0      [点我收藏+]

原文地址:

https://blog.csdn.net/C_chuxin/article/details/84990176

 

 

------------------------------------------------------------------------------------------

 

 

 

 

【题目】tf.ConfigProto和tf.GPUOptions用法总结

 

 

概述

tf.ConfigProto一般用在创建session的时候,用来对session进行参数配置,而tf.GPUOptions可以作为设置tf.ConfigProto时的一个参数选项,一般用于限制GPU资源的使用。

 

 

 

一、tf.ConfigProto

tf.ConfigProto一般用在创建session的时候,用来对session进行参数配置。

 

1.1 tf.ConfigProto()的参数

log_device_placement=True : 是否打印设备分配日志
 
allow_soft_placement=True : 如果你指定的设备不存在,允许TF自动分配设备

【转载】 tf.ConfigProto和tf.GPUOptions用法总结

原文:https://www.cnblogs.com/devilmaycry812839668/p/12034866.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!