Check failed: error == cudaSuccess (9 vs. 0) invalid configuration argument
请教一下，我训练了400张图片，测试100张图片，但是训练的时候出现了这样的错误，当我把batch_size调低时出现了out of memory的错误提醒。然后把batch_size调成1时出现了如上错误，怎么办，另外问一下调整batch_size时。是要两处都调整吗？
发起人
相关问题
- caffe装matlab接口报错Makefile:518: recipe for target 'matlab/+caffe/private/caffe_.mexa64' failed
- caffe源码中CHECK宏定义在哪个地方？
- 运行书中P102例程时提示 fatal error:cublas_v2.h no such file or direction
- Check failed: error == cudaSuccess (2 vs. 0) out of memory
- Check failed: img_height >= crop_size (0 vs. 227)
- VGG16训练的时候报错：loss_layer.cpp:19] Check failed: bottom[0]->num() == bottom[1]->num() (4 vs. 1)
- caffe如何使用data argument？
- 菜鸟，我在计算均值make_imagenet_mean.sh时出现了std::length_error'
- lmdb check failed MDB_BAD_TXN
- make runtest 显示ViennaCL: FATAL ERROR: Could not find kernel
问题状态
- 最新活动: 2017-04-10 23:56
- 浏览: 8936
- 关注: 5 人
3 个回复
phoenixbai
赞同来自: leemak 、木易ffenjoy
这个虽然是相同的错，但不同的layer因不同的原因都有可能会报这个错。
有的图片太大导致的，有的是参数太多导致的，都有可能。
所以，具体问题具体分析，要看挂在哪个layer上才能找到对应的解决方法。
豪的方法就是换显卡，必然解决了。不豪的，可能要调整网络结构中的参数来尝试了。
Evence
赞同来自: 木易ffenjoy
leemak
赞同来自:
还是显卡硬件的问题？