解决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
投稿猜你喜欢
- 如果要得到返回值，需要用Command的方法。 首先说明，返回值有两种。一种是在存储过程中直接return一个值，就象C和VB的函数返回值那
需要下载某网站的视频，chrome浏览器按F12打开开发者模式，发现视频链接是以"blob:http"开头的链接，打开这- 前言这部分已经折腾我两天了，还是没有头绪，可能还会折腾更久，最后在第三天上午解决问题，在一个不起眼的地方被坑了，jQuery加载的问题。会者
- 在win10环境下搭建python3.5.2和tensorflow平台，供大家参考，具体内容如下操作步骤如下：1、官网（https://ww
- 本文实例讲述了php使用Cookie实现和用户会话的方法。分享给大家供大家参考。具体分析如下：PHP 包含了很多的函数，可以用来管理和记录用
- 本文介绍了redis之django-redis的简单缓存使用，分享给大家，具体如下：自定义连接池这种方式跟普通py文件操作redis一样，代
- 前言超时，指一个协程A开启另一个协程B，A会阻塞等待B一段指定的时间，例如：5秒，A通知B结束（也有可能不通知，让B继续运行）。也就是说，A
- 通常python安装包都会被默认装在/usr/local/pythonx/lib/site-packages(linux)，但是我们有时想自
- 本文实例讲述了python使用正则表达式提取网页URL的方法。分享给大家供大家参考。具体实现方法如下：import reimport url
- 在平时，常常会提醒自己要做一个善于发现的人，时刻发现细节，这样才会更加懂得生活、会让生活更加精彩。在学习上也是如此，我们常常会发现有很多网站
- 当使用MySQL做站点的时候，肯定会有不知道的错误发生，怎么记录呢？以下是具体解决方法：class.method &n
- pyspark 操作hive表pyspark 操作hive表，hive分区表动态写入；最近发现spark动态写入hive分区，和saveAs
- 动手写个小组件(组件入门) 这篇文章主要是为想将自己的ASP水平提高的人写的! 把ASP代码变成组件,开发者不仅是加快了ASP的速度,而且也
- 之前看到过很多头条，说哪国某人坚持了多少年自学使用excel画画，效果十分惊艳。 对于他们的耐心我十分敬佩。 但是作为一个程序员，自然也得挑
- 本文实例讲述了PHP面向对象程序设计高级特性。分享给大家供大家参考，具体如下：静态属性<?phpclass StaticExample
- 影响 JavaScript性能的另外一个杀手就是递归，在上一节中提到采用memoization技术可以优化计算数值的递归函数，但memoiz
- virtualenv用于创建独立的Python环境，多个Python相互独立，互不影响，它能够：1. 在没有权限的情况下安装新套件2. 不同
- 一、说明：urllib库是python内置的一个http请求库，requests库就是基于该库开发出来的，虽然requests库使用更方便，
- 本文实例讲述了php隐藏IP地址后两位显示为星号的方法。分享给大家供大家参考。具体实现方法如下：我们在很多的公共网站中都会有碰到显示用户的I
- 本文实例讲述了Python使用迭代器捕获Generator返回值的方法。分享给大家供大家参考，具体如下：用for循环调用generator时
