RuntimeError: CUDA error:out of memory的一種解決辦法
具體環境:
Untubu:16.04
pytorch :1.5.0
CUDA:10.0
我是解決使用pytorch運行代碼時,遇到了如下錯誤:
RuntimeError: CUDA error:out of memory
我嘗試看網上的辦法很多方法,也沒有解決方法,解決然后想到之前運行過一篇類似的辦法代碼,其中好像有這樣的解決一行代碼:
然后我抱著嘗試的心態,在我的辦法代碼中加入下面倆行代碼:
import torch.backends.cudnn as cudnncudnn.benchmark = True
幸運的是,我的解決代碼可以正常運行了。當然這可能n種方法中的辦法一種,大家很是解決以自己的代碼的具體情況來修改自己的代碼。
辦法作者:Micia-0619
來源鏈接:https://www.cnblogs.com/macia/p/15523420.html