配置SSD-caffe测试时出现“Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal”解决
2017-11-03 11:25
1731 查看
这是由于GPU数量不匹配造成的,如果训练自己的数据,那么我们只需要将solver.prototxt文件中的device_id项改为自己的GPU块数,一块就是0,两块就是1,以此类推。
但是SSD配置时的例子是将训练语句整合成一个python文件ssd_pascal.py,所以需要改此代码。相关配置训练方法请参看转载博文:http://blog.csdn.net/xunan003/article/details/78427446
解决方法:将ssd_pascal.py文件中第332行gpus = "0,1,2,3"的GPU选择改为gpus = "0",后面的1,2,3都删掉即可。再次训练即可。
当然,由于博主只有一块GPU且电脑运行内存有限,还需要将ssd_pascal.py文件中的337行batch_size = 32和338行accum_batch_size = 32都改小一倍,即更改批量大小,不然会出现“Check failed: error == cudaSuccess (2 vs. 0) invalid ...”的错误。
总结:这两个错误只是由于博主硬件而出现的,所以请各位读者根据自己的硬件条件加以修正。
但是SSD配置时的例子是将训练语句整合成一个python文件ssd_pascal.py,所以需要改此代码。相关配置训练方法请参看转载博文:http://blog.csdn.net/xunan003/article/details/78427446
解决方法:将ssd_pascal.py文件中第332行gpus = "0,1,2,3"的GPU选择改为gpus = "0",后面的1,2,3都删掉即可。再次训练即可。
当然,由于博主只有一块GPU且电脑运行内存有限,还需要将ssd_pascal.py文件中的337行batch_size = 32和338行accum_batch_size = 32都改小一倍,即更改批量大小,不然会出现“Check failed: error == cudaSuccess (2 vs. 0) invalid ...”的错误。
总结:这两个错误只是由于博主硬件而出现的,所以请各位读者根据自己的硬件条件加以修正。
相关文章推荐
- 配置SSD-caffe测试时出现“Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal”解决
- caffe:Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal
- caffe 训练时,出现错误:Check failed: error == cudaSuccess (4 vs. 0) unspecified launch failure
- Check failed: error == cudaSuccess (8 vs. 0) invalid device function
- caffe运行错误: im2col.cu:61] Check failed: error == cudaSuccess (8 vs. 0) invalid device function
- caffe-windows10 安装问题 Check failed: error == cudaSuccess (8 vs.0) invalid device function
- Caffe 分类问题 Check failed: error == cudaSuccess (2 vs. 0) out of memory
- caffe 报错 Check failed: error == cudaSuccess (77 vs. 0) an illegal memory access was encountered
- Check failed: error == cudaSuccess (8 vs. 0) invalid device function
- caffe训练过程中显示Check failed:error == cudaSuccess(2 vs. 0) out of memory
- CUDA/caffe ERROR:cudaGetDeviceCount returned 30/35,Check failed: error == cudaSuccess (30/35 vs. 0)
- Check failed: error == cudaSuccess (8 vs. 0) invalid device function
- 【caffe跑试验遇到错误:Check failed: error == cudaSuccess (2 vs. 0) out of memory】:
- Check failed: error == cudaSuccess (8 vs. 0) invalid device function
- Caffe | Check failed: error == cudaSuccess (2 vs. 0) out of memory
- Check failed: error == cudaSuccess (8 vs. 0) invalid device function
- 【caffe】 Check failed: error == cudaSuccess (30 vs. 0) unknown error
- Check failed: error == cudaSuccess (8 vs. 0) invalid device function
- caffe在训练时遇到:Check failed: error == cudaSuccess (2 vs. 0) out of memory
- Check failed: error == cudaSuccess (8 vs. 0) invalid device function