学习随笔-ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[x,x]

今天跑代码的时候一直报这个ResourceExhaustedError,

刚开始我看后边提示shape[100,40960],

我以为是自己的网络参数设置错了,

但仔细检查之后运行还是同样报错,

我又认真就看了看报错问题,

发现是因为数据量太大,

我那一块显卡(GTX 1080)的显存(8G)不够用了,

于是我打开了终端.

输入:

watch -n 1 nvidia-smi

显示显卡的各种信息,

其中1代表数据刷新时间为1秒一次,

然后我看到python2.7占用了7G的显存,

可能是两天没关电脑的原因,

我赶紧关了pycharm,

当时显示显卡显存可用就增到了6G,

再次打开pycharm,

找到程序运行了一下,

运行成功,没有报错,

看来是时候加装显卡扩大显存了(以上来此梦里)

______________________________________

以下为来自2018年3月22日的更新

______________________________________

后来我发现,报这个资源用尽的错误有两种情况:

1. 训练数据量太大,网络参数太多,显卡显存不够用;

2. 在运行本次程序时,之前有程序还在运行,占用了显存.

综上来看,我遇到的应该是情况2,

关了pycharm这个ide就等于停止了所有的程序,

显存自然被释放了,

其实这样关掉ide是很不明智的做法,

正确操作应该是在pycharm界面右上角有一个红色方块stop键,

点击它stop了之前的程序就好了