解决TensorFlow GPU版出现OOM错误的问题
作者:lzher0 时间:2021-09-10 10:27:44
问题:
在使用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。
来源:https://blog.csdn.net/weixin_42007359/article/details/84634600
标签:TensorFlow,GPU,OOM
0
投稿
猜你喜欢
js倒记时代码,自定义时间改变时间,日期到时事件,日期格式。
2010-08-08 08:51:00
asp中通过addnew添加内容后取得当前文章的自递增ID的方法
2011-02-05 11:05:00
Microsoft VBScript 运行时错误 错误 800a0005 无效的过程调用或参数: chr
2011-03-09 11:03:00
如何用Python进行回归分析与相关分析
2023-11-24 01:08:14
交互设计实用指南系列(11)—减少记忆负担
2010-03-29 13:12:00
python 对任意数据和曲线进行拟合并求出函数表达式的三种解决方案
2021-10-02 12:40:16
PHP echo,print,printf,sprintf函数之间的区别与用法详解
2023-11-21 14:12:25
如何把数据库的记录输出到表格去?
2009-11-06 13:37:00
利用python实现周期财务统计可视化
2022-03-19 21:30:41
关于python爬虫应用urllib库作用分析
2023-11-02 12:59:43
Python使用scapy模块发包收包
2021-04-26 16:32:12
Python中unittest的数据驱动详解
2021-06-04 18:26:05
PHP PDOStatement::rowCount讲解
2023-06-06 12:24:04
PHP开发中常用的三个表单验证函数使用小结
2023-11-21 19:15:50
可以在线创建文件夹吗?
2009-11-01 18:07:00
用玩票的心态瞎猜豆瓣的思路
2008-08-18 21:14:00
PHP中Trait及其应用详解
2023-11-24 06:33:05
Python docutils文档编译过程方法解析
2023-01-20 16:41:23
详解python中文编码问题
2023-10-17 12:09:08
Python中 join() 函数的使用示例讲解
2023-03-29 02:32:25