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 限制显存大小的实现就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

Django实现登录随机验证码的示例代码
登录验证码是每个网站登录时的基本标配，这篇文章主要介绍了Django实现登录随机验证码的示例代码，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2018-06-06
pymongo实现控制mongodb中数字字段做加法的方法
这篇文章主要介绍了pymongo实现控制mongodb中数字字段做加法的方法,涉及Python使用pymongo模块操作mongodb数据库字段的技巧,具有一定参考借鉴价值,需要的朋友可以参考下
2015-03-03
一文带你了解Python 四种常见基础爬虫方法介绍
这篇文章主要介绍了一文带你了解Python 四种常见基础爬虫方法介绍，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-12-12
Python实战小项目之Mnist手写数字识别
MNIST 数据集已经是一个被”嚼烂”了的数据集, 很多教程都会对它”下手”, 几乎成为一个 “典范”. 不过有些人可能对它还不是很了解, 下面通过一个小实例来带你了解它
2021-10-10
Python中赋值的含义及说明
这篇文章主要介绍了Python中赋值的含义及说明，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2022-11-11
Python利用逻辑回归模型解决MNIST手写数字识别问题详解
这篇文章主要介绍了Python利用逻辑回归模型解决MNIST手写数字识别问题,结合实例形式详细分析了Python MNIST手写识别问题原理及逻辑回归模型解决MNIST手写识别问题相关操作技巧,需要的朋友可以参考下
2020-01-01
Python实现批量读取HDF多波段栅格数据并绘制像元直方图
这篇文章主要为大家详细介绍了如何基于Python语言gdal模块，实现多波段HDF栅格图像文件的读取、处理与像元值可视化（直方图绘制）等操作，需要的可以参考一下
2023-03-03
如何在Python函数执行前后增加额外的行为
有的时候会需要在函数前后添点额外的功能（比如过滤、计时等）时，以前总是首先想到装饰器。最近学习了Python的上下文管理器，所以本文就给大家介绍了如何在Python函数执行前后增加额外的行为，有需要的朋友们可以参考借鉴，下面来一起看看吧。
2016-10-10
Tensorflow 自带可视化Tensorboard使用方法(附项目代码)
这篇文章主要介绍了Tensorflow 自带可视化Tensorboard使用方法(附项目代码)，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2018-02-02
Pandas数据结构中Series属性详解
本文主要介绍了Pandas数据结构中Series属性详解，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2022-04-04