原文:RuntimeError: CUDA error:out of memory的一種解決辦法

具體環境: Untubu: . pytorch : . . CUDA: . 我是使用pytorch運行代碼時,遇到了如下錯誤: RuntimeError: CUDA error:out of memory 我嘗試看網上的很多方法,也沒有解決方法,然后想到之前運行過一篇類似的代碼,其中好像有這樣的一行代碼: 然后我抱着嘗試的心態,在我的代碼中加入下面倆行代碼: 幸運的是,我的代碼可以正常運行了。當然 ...

2021-11-08 11:25 0 849 推薦指數:

查看詳情

解決RuntimeError: CUDA error: out of memory

解決RuntimeError: CUDA error: out of memory 跑代碼時出現這個錯誤,輸入以下命令 $ nvidia-smi -l 原來是卡0內存爆了,於是指定相對空閑的2卡 可以正常運行 ...

Fri Mar 12 05:12:00 CST 2021 0 595
存在空閑CUDA前提下報錯:RuntimeError: CUDA error: out of memory

問題背景: 最近跑代碼時發現報錯CUDA out of memory,進入linux終端查看GPU使用情況(nvidia-smi),結果如下: 我用的GPU序號是0,但這塊被人占用了,所以我可以用剩下的3號和4號。 解決方案: 在代碼中更改GPU使用序號(修改/添加代碼 ...

Fri Sep 17 17:36:00 CST 2021 0 111
解決CUDA out of memory

問題 用Pytorch跑模型時,會出現RuntimeError: CUDA out of memory 解決方法 1.真的沒有內存了 可以改小batch_size 2.內存足夠,不分配 Win10下多線程的問題,num_threads/workers=0解決 ...

Wed Apr 08 19:18:00 CST 2020 0 6006
 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM