脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → 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错误的问题就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

相关文章

  • PyQt5事件处理之定时在控件上显示信息的代码

    PyQt5事件处理之定时在控件上显示信息的代码

    这篇文章主要介绍了PyQt5事件处理之定时在控件上显示信息的代码，本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2020-03-03
  • pytorch K折交叉验证过程说明及实现方式

    pytorch K折交叉验证过程说明及实现方式

    这篇文章主要介绍了pytorch K折交叉验证过程说明及实现方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
    2022-11-11
  • 手把手教你Python抓取数据并可视化

    手把手教你Python抓取数据并可视化

    很多小伙伴在提到python数据可视化的时候第一反应就是matplotlib库,但实际上python还有很多很好用的数据可视化的库,下面这篇文章主要给大家介绍了关于如何利用Python抓取数据并可视化的相关资料,需要的朋友可以参考下
    2022-05-05
  • 通过自学python能找到工作吗

    通过自学python能找到工作吗

    在本篇内容里小编给大家分享的是一篇关于通过自学python能找到工作吗的分析内容，对此有兴趣的朋友们可以参考下。
    2020-06-06
  • Python heapq库案例详解

    Python heapq库案例详解

    这篇文章主要介绍了Python heapq库案例详解,本篇文章通过简要的案例,讲解了该项技术的了解与使用,以下就是详细内容,需要的朋友可以参考下
    2021-09-09
  • Python正则表达式教程之三：贪婪/非贪婪特性

    Python正则表达式教程之三：贪婪/非贪婪特性

    这篇文章主要介绍了Python正则表达式中贪婪/非贪婪特性的相关资料，文中通过示例代码介绍的很详细，对大家具有一定的参考价值，需要的朋友下面来一起看看吧。
    2017-03-03
  • python 贪心算法的实现

    python 贪心算法的实现

    这篇文章主要介绍了python 贪心算法的实现，帮助大家更好的理解和学习python，感兴趣的朋友可以了解下
    2020-09-09
  • 详解Python开启线程和线程池的方法

    详解Python开启线程和线程池的方法

    这篇文章主要介绍了Python开启线程和线程池的方法，本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
    2024-03-03
  • Python基础知识之函数,类,模块

    Python基础知识之函数,类,模块

    今天小编就为大家分享一篇Python函数,类,模块，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2021-10-10
  • Python Web框架Flask中使用新浪SAE云存储实例

    Python Web框架Flask中使用新浪SAE云存储实例

    这篇文章主要介绍了Python Web框架Flask中使用新浪SAE云存储实例,本文是对SAE云存储的简单封装,需要的朋友可以参考下
    2015-02-02

最新评论