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 cudnn
cudnn.benchmark = True

幸运的是，我的代码可以正常运行了。当然这可能n种方法中的一种，大家很是以自己的代码的具体情况来修改自己的代码。

标签：代码,memory,cudnn,CUDA,error,RuntimeError,out
来源： https://www.cnblogs.com/macia/p/15523420.html