解决Tensorflow占用GPU显存问题
作者:xiaoxifei 时间:2021-09-01 14:34:46
我使用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
标签:Tensorflow,占用,GPU,显存
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
用python 绘制茎叶图和复合饼图
2023-08-04 10:34:54
![](https://img.aspxhome.com/file/2023/7/61457_0s.png)
python实现微信跳一跳辅助工具步骤详解
2023-08-02 11:11:40
go slice 数组和切片使用区别示例解析
2023-06-22 04:07:16
![](https://img.aspxhome.com/file/2023/0/87610_0s.jpg)
z-blog文章摘要图文混排
2009-02-28 13:49:00
![](https://img.aspxhome.com/file/UploadPic/20092/28/tagimg-64s.gif)
PHP json格式和js json格式 js跨域调用实现代码
2023-11-22 05:20:18
HTML和CSS中的视觉语义
2010-07-09 13:08:00
![](https://img.aspxhome.com/file/UploadPic/20107/9/01-27s.jpg)
超越MYSQL,ACCESS复合承载
2008-12-09 13:31:00
nginx简单配置多个php服务实例教程
2023-06-11 22:53:30
如何让对方在线查看我的程序代码?
2010-05-13 16:34:00
Python创建文件和追加文件内容实例
2021-03-15 04:38:09
php导出excel格式数据问题
2023-07-13 22:46:06
python 删除指定时间间隔之前的文件实例
2023-07-19 17:58:38
pytorch实现focal loss的两种方式小结
2023-07-02 14:43:22
Django 使用easy_thumbnails压缩上传的图片方法
2023-11-10 07:47:38
排序的人文魅力
2008-05-06 12:47:00
Golang并发编程之调度器初始化详解
2023-07-13 08:47:11
![](https://img.aspxhome.com/file/2023/5/99635_0s.png)
微信小程序利用Canvas绘制图片和竖排文字详解
2023-08-20 02:35:43
![](https://img.aspxhome.com/file/2023/2/56022_0s.jpg)
基于Python实现通过微信搜索功能查看谁把你删除了
2022-07-10 00:19:37
![](https://img.aspxhome.com/file/2023/2/79752_0s.jpg)
JavaScript 日期联动选择器
2010-08-01 10:18:00
最新LOGO设计流行趋势——叶子
2007-10-02 18:26:00
![](https://img.aspxhome.com/file/UploadPic/200710/2/2007102182921207s.jpg)