解决Tensorflow占用GPU显存问题
作者：xiaoxifei 发布时间：2021-09-01 14:34:46
标签：Tensorflow,占用,GPU,显存
我使用Pytorch进行模型训练时发现真正模型本身对于显存的占用并不明显，但是对应的转换为tensorflow后（权重也进行了转换），发现Python-tensorflow在使用时默认吃掉所有显存，并且不手动终结程序的话显存并不释放（我有两个序贯的模型，前面一个跑完后并不释放占用显存）（https://github.com/tensorflow/tensorflow/issues/1727），这一点对于后续的工作有很大的影响。
后面发现python-tensorflow限制显存有两种方法：
1. 设置显卡的使用率
这种方法在学习和工作中比较好用，学习时可提高显卡使用效率，工作时可方便的获得GPU显存消耗极限，用以提供显卡购买时的参数，现将代码展示如下：
这里的0.1 表示使用显存总量的的10％
2. 设置显卡按需使用（这个本人并没有专门测试，只是从tensorflow论坛上获得）
gpu_options = tf.GPUOptions(allow_growth=True)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
来源：https://blog.csdn.net/xiaoxifei/article/details/84337535
0
投稿猜你喜欢
对于一些数据量较大的系统，数据库面临的问题除了查询效率低下，还有就是数据入库时间长。特别像报表系统，每天花费在数据导入上的时间可能会长达几个- 前言官方手册:https://dev.mysql.com/doc/refman/5.7/en/server-logs.html不管是哪个数据
- 本文实例讲述了Python学习笔记之迭代器和生成器用法。分享给大家供大家参考，具体如下：迭代器和生成器迭代器每次可以返回一个对象元素的对象，
- 一、功能介绍在进行推广时，我们可以告诉对方，我们的微信公众账号是什么，客户可以去搜索，然后关注。二维码给我们提供了极大的便捷，只要简单一扫描
- asp无组件上传VBS编写的大家见的多了，这个是纯javascript实现的上传,原来unicode可以解决读取位置的问题，这次真的是纯JS
- 这篇文章主要介绍了pyftplib中文乱码问题解决方案,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋
- 文 | 李晓飞来源：Python 技术「ID: pythonall」最近有了一个新任务，需要将赛事视频，拆分成两分钟以内的小段，用
- 最近在学习机器学习的过程中，常常需要将本地写的代码传到GPU服务器中，然后在服务器上运行。之前的做法一直是先在本地写好代码，然后通过File
- vue+el使用this.$confirm不能阻断代码往下执行在vue+element ui的前端框架中使用el的confirm弹窗，遇到一
- 据小编统计绝大多数朋友对这一问题很棘手，今天小编给大家分享这一问题的解决方案，一起看看吧使用pip 提示更新：You are using p
- 二进制核心思想：冯诺依曼 + 图灵机电如何表示状态，才能稳定？计算机开始设计的时候并不是考虑简单，而是考虑能自动完成任务与结果的可靠性，简单
- 使用微信获取地址信息是和微信支付一道申请的，微信支付申请通过，就可以使用该功能。微信商城中，使用微信支付获取用户的收货地址，可以省略用户输入
- 一、INSERT语句中有反斜杠（\）1. 实际测试咱们用下面这些SQL来测试一下反斜杠（\）在INSERT语句中会是啥样？INSERT IN
- 今天搭了个“发短信”的页面，找朋友测试，没想到一位大侠直接弄了本长篇小说发我手机上……为了我的宝贝手机能继续健康澎湃，给文本区域(texta
- 由于客户提供的是excel文件，在使用时期望使用csv文件格式，且对某些字段内容需要做一些处理，如从某个字段中固定的几位抽取出来，独立作为一
- django-pipeline 是一个 Django 下非常方便的静态资源管理 app，尤其是 1.2 版本之后，利用 djan
- 本篇文章记录一下函数的等高线及其三维图像的可视化方法。本例绘制的函数为：1. 网格点  在绘制曲线之前，
- python除法负数商的取整方式与C++不同python：5 / -2 = -3若想和C++行为相同，可以使用 int(operator.t
- pyqtgraph官方给的示例居然会报错2333官方文档传送门：#####pyqtgraph exportpyqtgraph支持在可视化窗口
- 1. 安装 Git在 Windows 系统中安装Git非常简单，只需要下载Git的安装包，然后安装引导点击安装即可：Git下载地址：http
