tensorflow 限制显存大小的实现
更新时间：2020年02月03日 11:52:31 作者：安阳小栈-客官歇会吧
今天小编就为大家分享一篇tensorflow 限制显存大小的实现，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
Python在用GPU跑模型的时候最好开多进程，因为很明显这种任务就是计算密集型的。
用进程池好管理，但是tensorflow默认情况会最大占用显存，尽管该任务并不需要这么多，因此我们可以设置显存的按需获取，这样程序就不会死掉了。
1. 按比例预留：
tf_config = tensorflow.ConfigProto() tf_config.gpu_options.per_process_gpu_memory_fraction = 0.5 # 分配50% session = tensorflow.Session(config=tf_config)
2. 或者干脆自适应然后自动增长：
tf_config = tensorflow.ConfigProto() tf_config.gpu_options.allow_growth = True # 自适应 session = tensorflow.Session(config=tf_config)
以上这篇tensorflow 限制显存大小的实现就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
今天小编就为大家分享一篇numpy 进行数组拼接,分别在行和列上合并的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-05-05
Pytest+request+Allure实现接口自动化框架
接口自动化是指模拟程序接口层面的自动化，由于接口不易变更，维护成本更小，所以深受各大公司的喜爱，本文主要介绍了Pytest+request+Allure实现接口自动化框架，感兴趣的可以了解一下2021-07-07
这篇文章主要介绍了Python语言实现获取主机名根据端口杀死进程的相关资料,需要的朋友可以参考下2016-03-03
大家好，本篇文章主要讲的是用python实现谷歌小恐龙小游戏，看看这是你断网时的样子么，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下2022-02-02
这篇文章主要介绍了Python 并行化执行详细解析，文章围绕主题展开详细的内容介绍，具有一定的参考价值，需要的小伙伴可以参考一下，希望对你的学习有所帮助2022-07-07
这篇文章主要介绍了Python编程实现二分法和牛顿迭代法求平方根代码，具有一定参考价值，需要的朋友可以了解下。2017-12-12
今天小编就为大家分享一篇对python修改xml文件的节点值方法详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-12-12
这篇文章主要介绍了pycharm自定义TODO类注释以及高亮颜色的设置方法,文中通过图文结合的方式给大家介绍的非常详细,具有一定的参考价值,需要的朋友可以参考下2024-03-03
本例所登录的某网站需要提供用户名，密码和验证码，在此使用了python的urllib2直接登录网站并处理网站的Cookie2015-01-01
本文主要介绍了TensorFlow训练网络两种方式，一种是基于tensor(array)，另外一种是迭代器，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2021-12-12
最新评论