近期小编在做图像分类的实验,无奈状况连连。现将遇到的几个大坑简单记录一下,以备日后之需。如果你也出现了问题(1)中GPU内存问题,请看完整篇文章。
1、内存报错
如果你是GPU环境,那你要注意了,你的电脑上现在应该有两个内存限制。
(1)GPU内存
GPU内存大家可以使用下面命令查看
nvidia-smi
nvidia-smi -l
如果程序在运行过程中提示如下错误,那就说明是你的GPU内存不够了。
Check failed: error == cudaSuccess (2 vs. 0) out of memory