深度學習out of memory解決方案

2021-10-25 18:23:17 字數 398 閱讀 1783

跑程式時候出現 out of memory字樣

程式執行的資料記憶體超過了顯示卡專用gpu記憶體,導致記憶體溢位報錯。

import os

os.environ[

"cuda_visible_devices"]=

"1"

在程式首行加上上述命令,指定程式在哪個顯示卡上進行執行。我這裡有兩張顯示卡。所以最後的數字可以視情況,選擇「0」或者「1」。 如果兩張顯示卡都out of memory,最簡單粗暴的方式就是花錢再買一張顯示卡。或者是修改自己的程式,如深度學習中可以縮小batch size 的值,這樣也可以減少很大部分的記憶體使用。

Out of memory解決辦法

公司乙個專案,建立大規格lun的時候,隨著個數的增多,出現記憶體不足,程序被殺的情況,當時沒有多餘的記憶體條可以加,安照下面的方法可以暫時讓程序不被殺死。解決方法是 將swap大小設定大一點。從理論上來說,把swap大小設定為4g,那就不管實際記憶體多大,都不會有問題了,因為32位程式實際可訪問的虛...

Torch7 out of memory 解決方法

造成out of memory 的主要原因是lua預設的編譯器luajit的記憶體限制為2gb,解決的方法是用lua代替luajit,介紹了替換方法,具體操作如下 1.解除安裝之間安裝的torch 1.1 進入torch安裝路徑 clean.sh 1.2 刪除安裝的檔案 rm rf install....

孔徑合成演算法Out of memory的問題

out of memory 今天當我美滋滋地調好批量處理程式點下 start 處理昨天拍的還可以的細胞並且單獨phase都算好的時候,在孔徑合成的時候報錯了。心痛 昨天把角度增加到了49個方便做tomography,然而sa的演算法就記憶體不足了。確實為了提高解析度我也不能做降取樣,太難了 查了一些...