TensorFlow与显存

开始学习用TensorFlow,这东西与Theano不同,默认情况下, 在开启一个Session后,几乎占用显卡的所有显存。如果同一个机器、显卡多个人使用,基本上就是先到先得,后来的程序会崩溃。查了下文档有两种方法控制显存:

一、第一种是设置成预加载比例:

tf_config = tensorflow.ConfigProto()

tf_config.gpu_options.per_process_gpu_memory_fraction =0.5# 分配50%

session = tensorflow.Session(config=tf_config)

二、还有一种是自适应,需要多少就占多少:

tf_config = tensorflow.ConfigProto()

tf_config.gpu_options.allow_growth = True

session = tensorflow.Session(config=tf_config)

原文参考:http://friskit.me/2016/12/22/tensorflow-and-video-memory/

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容