RuntimeError: CUDA error: out of memory解决(亲测有效)

提前声明一下，我是在模型测试而不是模型训练时出现这个报错的，至于模型训练报此错误，请参考我的另一片博文：关于错误runtimeerror: CUDA out of memory. Tried to allocate 1.17 GB
其实解决方式很简单，原来我程序指定的gpu为3，运行测试代码时就报了标题out of memory的错误，如下图：

我将gpu指定为2和3，然后再运行代码就不会报错啦。

我又试了一下单独使用gpu为2时发现也不会报错，至于为什么指定3会报错而指定2不会报错，这个目前我还不太清楚，因为查看两个显卡，发现都没有使用，所以为了防止这类报错，索性就指定两个以上的gpu，一般情况下就可以解决了。