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