解決TensorFlow GPU版出現(xiàn)OOM錯誤的問題
更新時間：2020年02月03日 10:07:18 作者：lzher0
今天小編就為大家分享一篇解決TensorFlow GPU版出現(xiàn)OOM錯誤的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
問題：
在使用mask_rcnn預(yù)測自己的數(shù)據(jù)集時，會出現(xiàn)下面錯誤：
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，導(dǎo)致顯存不夠。
解決：
一是、將圖片尺寸改小，小到占用的內(nèi)存比顯存。
二是、不使用GPU進行預(yù)測，只使用CPU預(yù)測，因為一般CPU內(nèi)存要大于顯存的。但裝的又是GPU版的TensorFlow，所以需要在預(yù)測程序進行更改。
程序在前兩行加入下面代碼：
import os os.environ["CUDA_VISIBLE_DEVICES"] = ""
引號里填的是GPU的序號，不填的時候代表不使用GPU。
以上這篇解決TensorFlow GPU版出現(xiàn)OOM錯誤的問題就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了Python字符編碼判斷方法,結(jié)合實例形式分析了Python字符編碼的判斷技巧,并給出了chardet的安裝與使用方法,需要的朋友可以參考下2016-07-07
一年一度的六一兒童節(jié)又來了，祝大朋友小朋友節(jié)日快樂！本文主要介紹如何運用Python中的turtle庫控制函數(shù)繪制可達鴨，希望你會喜歡2022-05-05
今天小編就為大家分享一篇在Python 不同級目錄之間模塊的調(diào)用方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-01-01
這篇文章主要為大家詳細介紹了python多線程下信號處理程序示例，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-05-05
這篇文章主要介紹了Python圖形驗證碼識別，目前，許多網(wǎng)站采取各種各樣的措施來反爬蟲，其中一個措施便是使用驗證碼。隨著技術(shù)的發(fā)展，驗證碼的花樣越來越多。驗證碼最初是幾個數(shù)字組合的簡單的圖形驗證碼，后來加入了英文字母和混淆曲線2023-02-02
這篇文章主要為大家詳細介紹了如何使用python寫個喚醒睡眠電腦的腳本,文中的示例代碼講解詳細,感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下2025-01-01
一文搞懂Python的hasattr()、getattr()、setattr()?函數(shù)用法
python中的getattr()、setattr()、hasattr()函數(shù)均是對類屬性或方法的操作，其中g(shù)etattr()用于獲取類或?qū)嵗兄付ǚ椒ǐ@取屬性的值,setattr()用于設(shè)置類或?qū)嵗袑傩曰蚍椒?hasattr()用于判斷類或?qū)嵗惺欠翊嬖谥付ǖ膶傩曰蚍椒?本文通過例子給大家詳解，一起看看吧2022-04-04
這篇文章主要為大家詳細介紹了python地震數(shù)據(jù)可視化的相關(guān)資料，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-06-06
這篇文章主要為大家詳細介紹了Python如何實現(xiàn)將一個長圖制作成一個視頻，文中的示例代碼講解詳細，感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下2023-05-05
這篇文章主要介紹了python列表推導(dǎo)式，文章內(nèi)容附有經(jīng)典的詳細代碼，感興趣的小伙按可以參考下面文章具體內(nèi)容2021-10-10