脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → TensorFlow GPU版出现OOM错误

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

 更新时间：2020年02月03日 10:07:18   作者：lzher0  
今天小编就为大家分享一篇解决TensorFlow GPU版出现OOM错误的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

问题：

在使用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字符编码判断方法分析

    Python字符编码判断方法分析

    这篇文章主要介绍了Python字符编码判断方法,结合实例形式分析了Python字符编码的判断技巧,并给出了chardet的安装与使用方法,需要的朋友可以参考下
    2016-07-07
  • Python+Turtle绘制可爱的可达鸭

    Python+Turtle绘制可爱的可达鸭

    一年一度的六一儿童节又来了，祝大朋友小朋友节日快乐！本文主要介绍如何运用Python中的turtle库控制函数绘制可达鸭，希望你会喜欢
    2022-05-05
  • 在Python 不同级目录之间模块的调用方法

    在Python 不同级目录之间模块的调用方法

    今天小编就为大家分享一篇在Python 不同级目录之间模块的调用方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2019-01-01
  • python多线程下信号处理程序示例

    python多线程下信号处理程序示例

    这篇文章主要为大家详细介绍了python多线程下信号处理程序示例，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
    2019-05-05
  • Python图形验证码识别教程详解

    Python图形验证码识别教程详解

    这篇文章主要介绍了Python图形验证码识别，目前，许多网站采取各种各样的措施来反爬虫，其中一个措施便是使用验证码。随着技术的发展，验证码的花样越来越多。验证码最初是几个数字组合的简单的图形验证码，后来加入了英文字母和混淆曲线
    2023-02-02
  • python写个唤醒睡眠电脑的脚本

    python写个唤醒睡眠电脑的脚本

    这篇文章主要为大家详细介绍了如何使用python写个唤醒睡眠电脑的脚本,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下
    2025-01-01
  • 一文搞懂Python的hasattr()、getattr()、setattr() 函数用法

    一文搞懂Python的hasattr()、getattr()、setattr() 函数用法

    python中的getattr()、setattr()、hasattr()函数均是对类属性或方法的操作，其中getattr()用于获取类或实例中指定方法获取属性的值,setattr()用于设置类或实例中属性或方法,hasattr()用于判断类或实例中是否存在指定的属性或方法,本文通过例子给大家详解，一起看看吧
    2022-04-04
  • python地震数据可视化详解

    python地震数据可视化详解

    这篇文章主要为大家详细介绍了python地震数据可视化的相关资料，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
    2019-06-06
  • Python实现将长图制作成一个视频

    Python实现将长图制作成一个视频

    这篇文章主要为大家详细介绍了Python如何实现将一个长图制作成一个视频，文中的示例代码讲解详细，感兴趣的小伙伴可以跟随小编一起学习一下
    2023-05-05
  • python列表推导式 经典代码

    python列表推导式 经典代码

    这篇文章主要介绍了python列表推导式，文章内容附有经典的详细代码，感兴趣的小伙按可以参考下面文章具体内容
    2021-10-10

最新评论