GPU out of memory 問題 - Cupoy

2020/01/06 下午 05:33

電腦視覺深度學習討論版

胡連福

觀看數：1

回答數：1

收藏數：0

Build SSD_VGG.ipynb 跑 Use_cuda = False 時沒問題，

我試著把 Build SSD_VGG.ipynb 的 Use_cuda = True 設為 True，使用 GPU 來 run，但遇到以下out of memory問題 :

RuntimeError: CUDA out of memory. Tried to allocate 88.00 MiB (GPU 0; 2.00 GiB total capacity; 1.26 GiB already allocated; 17.00 MiB free; 19.23 MiB cached)

====================================

1. 我使用的 NB 內是用 Nvida-MX150 GPU，我用 nvidia-smi 查詢 memory 只顯示 2048 MiB(如圖一)，但我的系統 DRAM 已擴充到 16 GB(如圖二)，請問是哪裡的設定錯誤 ? 請專家告知應如何設定 ?

回答列表

• 2020/01/06 下午 11:37
  楊哲寧
  贊同數：0
  不贊同數：0
  留言數：2

  您好，這部分跟GPU的記憶體有關，跟Ram沒有關係，Nvida-MX150 GPU確實只有2G，跑影像相關模型時很容易會OOM，最快的解決辦法是調降batch size，如果batch size等於1還是會OOM的話，只能從模型結構下手。