解决TensorFlow GPU版出现OOM错误的问题


Posted in Python onFebruary 03, 2020

问题:

在使用mask_rcnn预测自己的数据集时,会出现下面错误:

ResourceExhaustedError: OOM when allocating tensor with shape[1,512,1120,1120] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
 [[{{node rpn_model/rpn_conv_shared/convolution}} = Conv2D[T=DT_FLOAT, data_format="NCHW", dilations=[1, 1, 1, 1], padding="SAME", strides=[1, 1, 1, 1], use_cudnn_on_gpu=true, _device="/job:localhost/replica:0/task:0/device:GPU:0"](fpn_p2/BiasAdd, rpn_conv_shared/kernel/read)]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.
 
 [[{{node roi_align_mask/strided_slice_17/_4277}} = _Recv[client_terminated=false, recv_device="/job:localhost/replica:0/task:0/device:CPU:0", send_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device_incarnation=1, tensor_name="edge_3068_roi_align_mask/strided_slice_17", tensor_type=DT_INT32, _device="/job:localhost/replica:0/task:0/device:CPU:0"]()]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

原因:

一是、因为图片尺寸为3200*4480,图片的尺寸太大。

二是、我使用的是TensorFlow GPU版,而我GPU的显存只有8G,导致显存不够。

解决:

一是、将图片尺寸改小,小到占用的内存比显存。

二是、不使用GPU进行预测,只使用CPU预测,因为一般CPU内存要大于显存的。但装的又是GPU版的TensorFlow,所以需要在预测程序进行更改。

程序在前两行加入下面代码:

import os
os.environ["CUDA_VISIBLE_DEVICES"] = ""

引号里填的是GPU的序号,不填的时候代表不使用GPU。

以上这篇解决TensorFlow GPU版出现OOM错误的问题就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python3中常用的处理时间和实现定时任务的方法的介绍
Apr 07 Python
在Mac OS系统上安装Python的Pillow库的教程
Nov 20 Python
浅谈Python 中整型对象的存储问题
May 16 Python
python利用正则表达式搜索单词示例代码
Sep 24 Python
浅谈dataframe中更改列属性的方法
Jul 10 Python
详解Python3注释知识点
Feb 19 Python
Python Django 前后端分离 API的方法
Aug 28 Python
Python3爬虫里关于识别微博宫格验证码的知识点详解
Jul 30 Python
python Paramiko使用示例
Sep 21 Python
Pycharm安装python库的方法
Nov 24 Python
Python django中如何使用restful框架
Jun 23 Python
Python如何加载模型并查看网络
Jul 15 Python
python global和nonlocal用法解析
Feb 03 #Python
Tensorflow设置显存自适应,显存比例的操作
Feb 03 #Python
解决Tensorflow占用GPU显存问题
Feb 03 #Python
通过Turtle库在Python中绘制一个鼠年福鼠
Feb 03 #Python
python爬虫模块URL管理器模块用法解析
Feb 03 #Python
Tensorflow实现多GPU并行方式
Feb 03 #Python
python如何通过twisted搭建socket服务
Feb 03 #Python
You might like
PHP查询快递信息的方法
2015/03/07 PHP
PHP实现根据图片色界在不同位置加水印的方法
2015/08/08 PHP
php框架知识点的整理和补充
2021/03/01 PHP
js loading加载效果实现代码
2009/11/24 Javascript
jQuery实现密保互斥问题解决方案
2013/08/16 Javascript
jQuery中add()方法用法实例
2015/01/08 Javascript
javascript中AJAX用法实例分析
2015/01/30 Javascript
js完美实现@提到好友特效(兼容各大浏览器)
2015/03/16 Javascript
JavaScript整除运算函数ceil和floor的区别分析
2015/04/14 Javascript
浅析javascript中的事件代理
2015/11/06 Javascript
javascript数据结构之二叉搜索树实现方法
2015/11/25 Javascript
使用vuex的state状态对象的5种方式
2018/04/19 Javascript
JavaScript实现仿Clock ISO时钟
2018/06/29 Javascript
vue实现微信二次分享以及自定义分享的示例
2019/03/20 Javascript
五分钟搞懂Vuex实用知识(小结)
2019/08/12 Javascript
在博客园博文中添加自定义右键菜单的方法详解
2020/02/05 Javascript
JavaScript HTML DOM 元素 (节点)新增,编辑,删除操作实例分析
2020/03/02 Javascript
python下如何让web元素的生成更简单的分析
2008/07/17 Python
python文件操作相关知识点总结整理
2016/02/22 Python
python定向爬取淘宝商品价格
2018/02/27 Python
用python爬取租房网站信息的代码
2018/12/14 Python
对Python3中dict.keys()转换成list类型的方法详解
2019/02/03 Python
python requests证书问题解决
2019/09/05 Python
Django框架创建项目的方法入门教程
2019/11/04 Python
使用darknet框架的imagenet数据分类预训练操作
2020/07/07 Python
最好的意大利皮夹克:D’Arienzo
2018/12/04 全球购物
西班牙最大的在线滑板和街头服饰商店:Fillow.net
2019/04/15 全球购物
阿迪达斯英国官方网站:adidas英国
2019/08/13 全球购物
数控专业个人求职信范文
2014/02/05 职场文书
2014年小学元旦活动方案
2014/02/12 职场文书
优秀实习生主要事迹
2014/05/29 职场文书
授权委托书样本
2014/09/25 职场文书
作风建设剖析材料
2014/10/06 职场文书
2015年秋季运动会加油稿
2015/07/22 职场文书
浅谈resultMap的用法及关联结果集映射
2021/06/30 Java/Android
Python jiaba库的使用详解
2021/11/23 Python