解决Tensorflow占用GPU显存问题
更新时间：2020年02月03日 09:43:43 作者：xiaoxifei
今天小编就为大家分享一篇解决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))
以上这篇解决Tensorflow占用GPU显存问题就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
今天小编就为大家分享一篇对python中for 、if、 while的区别与比较方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-06-06
word、excel、PPT，虽然说是特殊文件，其实也是实际工作中我们经常会用到的文件类型。本文将为大家详解Python读取Word文件和文件内容的方法，感兴趣的可以了解一下2022-05-05
这篇文章主要介绍了浅谈python 线程池threadpool之实现，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2017-11-11
这篇文章主要介绍了ORM Django 终端打印 SQL 语句实现解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-08-08
这篇文章主要为大家介绍了Python光学仿真wxpython透镜演示系统计算与绘图的实现示例。有需要的朋友可以借鉴参考下，希望能够有所帮助2021-10-10
这篇文章主要介绍了Python的单元测试,代码基于Python2.x版本,需要的朋友可以参考下2015-04-04
Scrapy-Redis之RedisSpider与RedisCrawlSpider详解
这篇文章主要介绍了Scrapy-Redis之RedisSpider与RedisCrawlSpider详解，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-11-11
这篇文章主要介绍了python Tkinter的简单入门教程，帮助大家更好的理解和学习使用python制作gui程序，感兴趣的朋友可以了解下2021-04-04
这篇文章主要介绍了Python基于Tkinter编写crc校验工具，本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下2020-05-05
这篇文章主要介绍了Python调用腾讯API进行人脸身份证比对,简单介绍了调用腾讯云API步骤，通过完整代码展示与结果，需要的朋友可以参考下2022-04-04
最新评论