【Tensorflow】设置显存自适应,显存比例

用惯了theano.再用tensoflow发现一运行显存就满载了,吓得我吃了一个苹果。 
用天朝搜索引擎毛都搜不到,于是FQ找了下问题的解决方法,原来有两种 

 按比例

config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.4 session = tf.Session(config=config, ...)

按需求增长(theano那种)

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config, ...)
全部评论

相关推荐

程序员花海:实习和校招简历正确格式应该是教育背景+实习+项目经历+个人评价 其中项目经历注意要体现业务 实习经历里面的业务更是要自圆其说 简历模板尽可能保持干净整洁 不要太花哨的
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务