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 限制显存大小的实现就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
Python bsddb模块操作Berkeley DB数据库介绍
这篇文章主要介绍了Python bsddb模块操作Berkeley DB数据库介绍,这里简单介绍一些关于bsddb的使用方法,需要的朋友可以参考下2015-04-04
这篇文章主要介绍了pycharm显示行号的方法,需要的朋友可以参考下2014-06-06
python UPX is not available问题解决方法
这篇文章主要介绍了python UPX is not available问题解决,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2023-04-04
这篇文章主要给大家介绍了关于Python3基础教程之递归函数简单示例的相关资料，文中通过示例代码介绍的非常详细，对大家学习或者使用Python3具有一定的参考学习价值，需要的朋友们下面来一起学习学习吧2019-06-06
这篇文章主要为大家详细介绍了opencv实现图片模糊和锐化操作，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-11-11
这篇文章主要介绍了python制作一个解压缩软件的方法，非常不错，具有一定的参考借鉴价值,需要的朋友可以参考下2019-11-11
python登录QQ邮箱发信的代码，有需要的朋友可以参考下2013-02-02
这篇文章主要介绍了Python 如何实现变量交换，Python 程序员肯定知道 a,b = b,a，这句话用来交换两个变量。相较于其它语言需要引入一个 temp 来临时存储变量的做法，Python 的这种写法无疑非常优雅，下面我们来看看具体的实现过程吧2022-01-01
这篇文章主要介绍了详解python中各种的函数,是Python入门中的基础知识,需要的朋友可以参考下，希望能够给你带来帮助2021-10-10
python字符串分割常用方法(str.split()和正则)
在Python中字符串是一种非常常见的数据类型,在实际应用中我们经常需要对字符串进行分割,以便对其中的内容进行处理,这篇文章主要给大家介绍了关于python字符串分割(str.split()和正则)的相关资料,需要的朋友可以参考下2023-11-11
最新评论